将 DataFrame 中的每个唯一值重塑为列

Question

我有一个与此类似的 DataFrame：

  12 11 10
1  a  b  a
2  c  c  a
3  c  b  b
4  b  a  a

我想为 DataFrame 中的每个唯一值创建一个列，并将这些列作为值，例如：

         a        b        c
1  [12,10]     [11]      NaN
2     [10]      NaN  [12,11]
3      NaN  [11,10]     [12]
4  [11,10]     [12]      NaN

Answer 1

您可以通过DataFrame.stack重塑，将MultiIndex转换为列，然后使用Series.unstack聚合list ，最后通过DataFrame.rename_axis删除索引和列名称：

df1 = (df.stack()
         .reset_index(name='v')
         .groupby(['level_0','v'])['level_1']
         .apply(list)
         .unstack()
         .rename_axis(index=None, columns=None))
print (df1)
          a         b         c
1  [12, 10]      [11]       NaN
2      [10]       NaN  [12, 11]
3       NaN  [11, 10]      [12]
4  [11, 10]      [12]       NaN

或者将索引转换为列，因此可以通过它使用DataFrame.melt然后使用与以前相同的步骤：

df1 = (df.reset_index()
         .melt('index')
         .groupby(['index','value'])['variable']
         .apply(list)
         .unstack()
         .rename_axis(index=None, columns=None))
print (df1)
          a         b         c
1  [12, 10]      [11]       NaN
2      [10]       NaN  [12, 11]
3       NaN  [11, 10]      [12]
4  [11, 10]      [12]       NaN