假设数据存储在一个名为data
的DataFrame中,包含两列A和B。
import pandas as pd
# 创建示例数据
data = {'A': [1, 2, 3, 1, 2, 3, 1, 2, 3],
'B': ['a', 'a', 'a', 'b', 'b', 'b', 'c', 'c', 'c']}
df = pd.DataFrame(data)
# 按照B列分组,计算A列中不同的值的数量
result = df.groupby('B')['A'].nunique()
print(result)
输出结果如下:
B
a 3
b 2
c 1
Name: A, dtype: int64
以上代码首先导入了pandas
库,然后创建了一个包含两列A和B的DataFrame。接着使用groupby
函数按照B列进行分组,并使用nunique
函数计算每个分组中A列中不同的值的数量。最后,打印输出结果。