python - Pandas DataFrame 的异常 reshape

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

我需要以保持“x”列唯一的方式 reshape 它:

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

所以 'y_N' 列的最大 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())