根据熊猫中的多个条件分组[重复]

Question

我想对两列使用不同的条件来聚合行。

当我做df时。groupby（“[a]”）。agg（'count'），我得到输出1

当我做df.groupby（'[a]'）. agg（'均值'）时，我得到输出2

是否有一种方法可以进行聚合，将输出1显示到列[b]，将输出2显示到列[c]？

Answer 1

下面的代码应该有效：

# Import libraries
import pandas as pd
import numpy as np

# Create sample dataframe
df = pd.DataFrame({'a': ['A1', 'A1', 'A2', 'A3', 'A4', 'A3'],
                   'value': [1,2,3,4,5,6]})

# Calculate count, mean 
temp1 = df.groupby(['a']).count().reset_index().rename(columns={'value':'count'})
temp2 = df.groupby(['a'])['value'].mean().reset_index().rename(columns={'value':'mean'})

# Add columns to existing dataframe
df.merge(temp1, on='a', how='inner').merge(temp2, on='a', how='inner')

# Add columns to a new dataframe
df2 = temp1.merge(temp2, on='a', how='inner')
df2