pandas - 将多列合并为一列

Question

我想从这个结构中融化我的 pandas 数据框......

<表类=“s-表”> <标题> 姓名 2020 色彩年 2019年色彩年 2020 年规模 2019 年规模 <正文> 一个 红色 红色 大 小 B 红色 蓝色 大 中

...到这个。

<表类=“s-表”> <标题> 姓名 年 颜色 尺寸 <正文> 一个 2020 红色 大 B 2020 红色 大 一个 2019 红色 小 B 2019 蓝色 中

我试图融化

df_melted = df.melt(
    id_vars=['Name', 'Color year 2020', 'Color year 2019'],
    value_vars=['Size year 2020', 'Size year 2019'],
    var_name='Year',
    value_name='Size')

df_final = df_melted.melt(
    id_vars=['Name', 'Year'],
    value_vars=['Color year 2020', 'Color year 2019'],
    var_name='Year',
    value_name='Color')

但这似乎不对......是否可以用一种方式做到这一点？

Answer 1

您可以使用wide_to_long进行检查

out = pd.wide_to_long(df, ['Color year ','Size year '],
                  i = 'Name', j ='Year',suffix = '\w+')
out.columns = out.columns.str.split().str[0]
out.reset_index(inplace=True)
out
Out[151]: 
  Name  Year Color    Size
0    A  2020   red     big
1    B  2020   red     big
2    A  2019   red   small
3    B  2019  blue  medium