熊猫 groupby 和 qcut

Posted 2023-03-11

【中文标题】熊猫 groupby 和 qcut

【英文标题】：Pandas groupby and qcut

【发布时间】：2013-10-24 13:13:06

【问题描述】：

有没有办法构建 Pandas 的 groupby 和 qcut 命令以返回包含嵌套图块的列？具体来说，假设我有 2 组数据，我希望将 qcut 应用于每组，然后将输出返回到一列。这类似于 MS SQL Server 的 ntile() 命令，它允许通过 () 进行分区。

     A    B  C
0  foo  0.1  1
1  foo  0.5  2
2  foo  1.0  3
3  bar  0.1  1
4  bar  0.5  2
5  bar  1.0  3

在上面的数据框中，我想将 qcut 函数应用于 B，同时在 A 上进行分区以返回 C。

【参考方案1】：

import pandas as pd
df = pd.DataFrame('A':'foo foo foo bar bar bar'.split(),
                   'B':[0.1, 0.5, 1.0]*2)

df['C'] = df.groupby(['A'])['B'].transform(
                     lambda x: pd.qcut(x, 3, labels=range(1,4)))
print(df)

产量

     A    B  C
0  foo  0.1  1
1  foo  0.5  2
2  foo  1.0  3
3  bar  0.1  1
4  bar  0.5  2
5  bar  1.0  3