Pandas：获取假人并将其他列保留为值

Question

我有一个这样的 dataframe：

顺序	信息
1个	一种
2个	乙
3个	C
4个	丁
1个	一种
2个	C
3个	丁
4个	乙
5个	F
1个	一种
2个	C
3个	丁

我正在尝试 pivot 以最大Sequence长度作为列并将相应的可能Message作为值的表。

1个	2个	3个	4个	5个	6个	7
一种	乙	C	丁
一种	C	丁	乙	F
一种	C	丁

我正在尝试将 pd.get_dummies() 用于Sequence ，但无法弄清楚如何保留Message值。

感谢任何帮助。

Answer 1

一个可能的解决方案：

(df.groupby(df['Sequence'].where(df['Sequence'] == 1, 0).cumsum())
 .apply(lambda x: x.set_index('Sequence').T)
 .reset_index(drop=True).rename_axis(None, axis=1))

Output：

   1  2  3    4    5
0  A  B  C    D  NaN
1  A  C  D    E    F
2  A  C  D  NaN  NaN

Answer 2

使用pivot和df['Sequence'].diff().le(0).cumsum()作为索引（当序列重新启动到较早的数字时的新行）：

(df.assign(index=df['Sequence'].diff().le(0).cumsum())
   .pivot(index='index', columns='Sequence', values='Message')
   .rename_axis(index=None, columns=None) # optional
)

Output：

   1  2  3    4    5
0  A  B  C    D  NaN
1  A  C  D    E    F
2  A  C  D  NaN  NaN