在pandas中,可以使用groupby
函数对不同的行进行分组,并将类别的计数添加到新列中。
首先,需要导入pandas库:
import pandas as pd
然后,创建一个包含类别数据的DataFrame:
data = {'Category': ['A', 'B', 'A', 'B', 'A', 'B']}
df = pd.DataFrame(data)
接下来,使用groupby
函数按照类别进行分组,并使用size
函数计算每个类别的计数:
df['Count'] = df.groupby('Category')['Category'].transform('size')
最后,可以打印输出DataFrame来查看结果:
print(df)
输出结果如下:
Category Count
0 A 3
1 B 3
2 A 3
3 B 3
4 A 3
5 B 3
在这个例子中,我们按照类别对行进行了分组,并将每个类别的计数添加到了新列"Count"中。
领取专属 10元无门槛券
手把手带您无忧上云