为datafram中的条件组创建列

Question

我在数据帧df中有一个表

id   key_no
1     1
2     1
3     2
4     2
5     2
6     3
7     3

在此特定的密钥中，编号与多个id相关联

我想创建一个包含列的新数据帧

keyno    start_id    end_id
 1          1          2
 2          3          5
 3          6          7

即在数据帧df2中为每个keyno创建列“start\u id”和“end\u id”

我们能试着用吗数据框groupby，但是如何使用它创建新的df2，我对python还不熟悉， 有线索吗？你知道吗

Answer 1

通过^{}和^{}使用^{}+^{}。最后rename列按dict：

d = {'first':'start_id','last':'end_id'}
df = df.groupby('key_no')['id'].agg(['first','last']).rename(columns=d)
print (df)
        start_id  end_id
key_no                  
1              1       2
2              3       5
3              6       7