使用 Python 查找均值、众数和中位数

Question

我正在尝试计算数据框中 2 列的平均值、众数和中位数。我想按 COL1 分组并计算 COL2 的平均值、众数和中位数。有许多列和行，但我希望使用的 2 个示例如下所示

    COL1      COL2    
1   (0-9)      64  
2   (10-19)    74   
3   (20-29)    64
4   (0-9)      62

我一直在尝试以下代码，但不断出错：

data.groupby('COL1')['COL2'].mean()

如果我只使用 COL1 进行分组，这将返回其他 2 列的平均值（和中位数，但不是众数），但不是 COL2：

data1 = data.groupby(['COL1']).mean()

我认为 COL2 显示为对象可能存在问题，但我很难解决这个问题。示例错误如下所示：

pandas.core.base.DataError：没有要聚合的数字类型

Answer 1

您可以将类型设置为int，然后使用pd.Series.mode 计算模式（以及平均值和中位数）

df = df[df.notna()] #Handling nans
df['COL2'] = df['COL2'].astype(float)

mode = pd.Series.mode

#INCASE OF ValueError: Must produce aggregated value, Use this for mode
#mode = lambda x: pd.Series.mean(pd.Series.mode(x))

out = df.groupby("COL1")["COL2"].agg(['mean','median', mode])
print(out)

如果一个组没有单一模式，您可以取pd.Series.mode 返回的模式的平均值。只需将模式替换为注释的 lambda 函数即可。

单个值的平均值不会改变众数。随意将函数更改为您对模式的定义，以防找不到单一模式。