pandas 添加一个基于分组的排序列

Question

假设有一个包含两列的数据框，其中 col1 表示组。

d = pd.DataFrame({'col1': ['a','a','a','a', 'a', 'b','b'], 'col2': ['nmh','ghb','dfe', 'dfe', 'kil', 'gtr','klm']})

我想添加第三列，它使用 col1 中的组和 col2 中的条目，并添加线性顺序，如下所示：

order = [1,2,3,3,4, 1,2]
d['order'] = order
d

col2 将大部分是唯一的，如果有任何内容重复 order 列应该重复订单号。

我用了groupby和rank都没有用。正常情况下提供method='first'到rank方法应该可以解决问题，但是报错。

注意：col1中每个组对应的条目数不同，df会更大。所以请提供一个通用的答案。

Answer 1

In [45]: d['order'] = (d.groupby('col1')['col2']
                        .transform(lambda x: (x!=x.shift()).cumsum()))

In [46]: d
Out[46]:
  col1 col2  order
0    a  nmh      1
1    a  ghb      2
2    a  dfe      3
3    a  dfe      3
4    a  kil      4
5    b  gtr      1
6    b  klm      2

或来自@Zero 的更好替代方案：

In [52]: d.col2.ne(d.col2.shift()).groupby(d.col1).cumsum()
Out[52]:
0    1.0
1    2.0
2    3.0
3    3.0
4    4.0
5    1.0
6    2.0
Name: col2, dtype: float64

Answer 2

使用factorize

d['Order']=d.groupby('col1').col2.transform(lambda x : pd.factorize(x)[0]+1)
d
Out[1641]: 
  col1 col2  Order
0    a  nmh      1
1    a  ghb      2
2    a  dfe      3
3    a  dfe      3
4    a  kil      4
5    b  gtr      1
6    b  klm      2

pandas 添加一个基于分组的排序列

pandas add an order column based on grouping

sorting

rank

pandas

pandas-groupby