需求

image.png

分析：

1. 一行变多行，可以用explode实现；
2. 要使用explode，需要先将多列变成一列；
3. 注意有的列为空，需要做空值过滤；

数据准备

file_path = "/course_datas/c39_explode_to_manyrows/读者提供的数据-输入.xlsx"
df = pd.read_excel(file_path,engine = "openpyxl")
print(df.head())

image.png

多列合并到一列

merge_names = list(df.loc[:,"Supplier":].columns.values)
print("merge_names","\n",merge_names)

def merge_cols(x):
    """
    x是一个行Serise，把他们按分隔符合并
    :param x:
    :return:
    """
    # 删除为空的列
    x = x[x.notna()]
    print("x","\n",x)
    #使用x.values用于合并
    y = x.values
    print("y","\n",y)
    # 合并后的列表，每个元素是"Supplier" + "Supplier PN"对
    result = []
    # range的步长为2，目的是每两列做合并
    for idx in range(0, len(y), 2):
        # 使用竖线作为"Supplier" + "Supplier PN"之间的分隔符
        result.append(f"{y[idx]}|{y[idx+1]}")
    # 将所有两两对，用#分割，返回一个大字符串
    return "#".join(result)

df["merge"] = df.loc[:,"Supplier":].apply(merge_cols,axis = 1)
print(df)

image.png

运行结果

删除不需要的列

df.drop(merge_names,axis = 1,inplace=True)
print(df)

image.png

使用explode把一列变多行

#先将merge列变成list的形式
df["merge"] = df["merge"].str.split("#")
# print(df)

image.png

执行explode变成多行
df_explode = df.explode("merge")
print(df_explode)

image.png

将一列还原成结果的多列

# 分别从merge中提取两列
df_explode["Supplier"]=df_explode["merge"].str.split("|").str[0]
df_explode["Supplier PN"]=df_explode["merge"].str.split("|").str[1]
print(df_explode)

# 把merge列删除掉，得到最终数据
df_explode.drop("merge", axis=1, inplace=True)
print(df_explode)

image.png