在pandas dataframe中为条件创建列

Question

我在pandas dataframe df中有一个表

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

在此特定key_no中与多个id

相关联

我想创建一个包含列

的新数据框

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

即创建列＆quot; start_id＆＃39;和＆＃39; end_id＆＃39;对于每个keyno，在dataframe df2

中

我们可以尝试使用df.groupby，但是如何使用它创建新的df2，我是python的新手， 任何线索？

Answer 1

groupby和agg使用first + last。 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