如何获得整个数据框而不是列的均值?

时间:2018-06-22 10:25:57

标签: python pandas dataframe mean

如何获取熊猫数据框中所有值的平均值(NaN除外)?

pd.DataFrame.mean()仅给出了每一列(或设置axis=1时的行)的均值,但是我想要整个事情的均值。而且df.mean().mean()并不是最明智的选择(请参见下文)。

请注意,在我的实际情况下,数据框具有较大的多重索引,这使事情更加复杂。对于无关紧要的情况,可以将@EdChum的答案视为更直接的方法,在某些情况下,它可能比更快的解决方案更可取。

示例代码

data1 = np.arange(16).reshape(4, 4)
df = pd.DataFrame(data=data1)

df.mean()
0    9.0
1    7.0
2    8.0
3    9.0
dtype: float64

df.mean().mean()
7.5

np.arange(16).mean()
7.5

有效,但是如果我掩盖了df的一部分(实际上,它是数百行/列的相关矩阵,从本质上说,它本身的一半充满了冗余数据),

triang = np.triu_indices(4)
data2 = np.arange(4.,20.).reshape(4, 4)
data2[triang]=np.nan
df2 = pd.DataFrame(data=data2)

df2.mean().mean()
15.0

但是(8. + 12. + 13. + 16. + 17. + 18.)/614.

除了编写手动执行上述操作的某种循环之外,如何最好地获得“真实”的意思?

2 个答案:

答案 0 :(得分:4)

您可以使用numpy.nanmean

triang = np.triu_indices(4)
data2 = np.arange(4.,20.).reshape(4, 4)
data2[triang]=np.nan
df2 = pd.DataFrame(data=data2)

res = np.nanmean(df2)  # 14.0

也可以通过stack,如@EdChum所述,但速度较慢:

df2 = pd.concat([df2]*100000)

%timeit np.nanmean(df2)              # 14.0ms
%timeit df2.stack().dropna().mean()  # 55.7ms

如果您的数据仅是数字数据,则还可以完全删除Pandas的开销。

答案 1 :(得分:2)

您可以先stack,然后dropna,然后致电mean

In[201]:
df2.stack().dropna().mean()

Out[201]: 14.0

因此,这会将df转换为单列Series,然后您可以调用dropna删除NaN行,现在mean将计算{{1} }。