pandas分组运算(groupby)

1. groupby()

import pandas as pd
df = pd.DataFrame([[1, 1, 2], [1, 2, 3], [2, 3, 4]], columns=["A", "B", "C"])
print(df)

g = df.groupby('A').mean()   # 按A列分组(groupby),获取其他列的均值
print(g)

# 方法1
b = df['B'].groupby(df['A']).mean()    # 按A列分组,获取B列的均值
print(b)

# 方法2
b = df.ix[:,1].groupby(df.ix[:, 0]).mean()    # 按A列分组(0对应A列,1对应B列),获取B列的均值
print(b)

# 方法3
m = df.groupby('A')
b = m['B'].mean()
print(b)

2. 聚合方法size()和count()

size跟count的区别: size计数时包含NaN值,而count不包含NaN值

import pandas as pd 
import numpy as np

df = pd.DataFrame({"Name":["Alice", "Bob", "Mallory", "Mallory", "Bob" , "Mallory"],
                   "City":["Seattle", "Seattle", "Portland", "Seattle", "Seattle", "Portland"],
                   "Val":[4,3,3,np.nan,np.nan,4]})
print(df)

count()

a = df.groupby(["Name", "City"], as_index=False)['Val'].count()
print(a)

size()

b = df.groupby(["Name", "City"])['Val'].size().reset_index(name='Size')
print(b)

来自:https://blog.csdn.net/m0_37870649/article/details/80979809

原文地址:https://www.cnblogs.com/keye/p/11153427.html