Pandas - 基于组和列值有效地连接行

Question

假设我们有一个具有以下结构的 pandas 数据框： Id | number |seq| a | 1 | X | a | 3 | Z | b | 1 | P | b | 6 | G | b | 5 | C |

所需的结果是一个如下所示的数据框： Id |new_seq| a | XZ | b | PCG |

即new_seq列是seq按Id分组，并按照number列指示的顺序。

有哪些选项可以有效地实施此操作？

Answer 1

您可以将sort_values 与groupby 和apply 一起使用join：

print (df.sort_values(['Id', 'number']))
  Id  number seq
0  a       1   X
1  a       3   Z
2  b       1   P
4  b       5   C
3  b       6   G

df = df.sort_values(['Id', 'number'])
       .groupby('Id', sort=False).seq
       .apply(''.join)
       .reset_index(name='new_seq')

print (df)
  Id new_seq
0  a      XZ
1  b     PCG