不寻常的Pandas DataFrame重塑

Question

我有这样的DF：

df = pd.DataFrame({'x': ['a', 'a', 'b', 'b', 'b', 'c'],
                   'y': [1, 2, 3, 4, 5, 6],
                 })

看起来像：

   x  y
0  a  1
1  a  2
2  b  3
3  b  4
4  b  5
5  c  6

我需要重塑它以保持＆＃39; x＆＃39;列唯一：

   x    y_1  y_2  y_3
0  a    1    2    NaN
1  b    3    4    5
2  c    6    NaN  NaN

所以＆＃39; y_N＆＃39;的最大N列必须等于

max(df.groupby('x').count().values)

并且x列必须包含唯一值。

现在我不知道如何获得y_N列。

感谢。

Answer 1

您可以将pandas.crosstab与cumcount列一起用作columns参数：

(pd.crosstab(df.x, df.groupby('x').cumcount() + 1, df.y, 
            aggfunc = lambda x: x.iloc[0])
   .rename(columns="y_{}".format).reset_index())