python numpy ndarray元素明智的意思

Question

我想计算numpy ndarray的元素平均值.

In [56]: a = np.array([10, 20, 30])

In [57]: b = np.array([30, 20, 20])

In [58]: c = np.array([50, 20, 40])

我想要的是:

[30, 20, 30]

除了矢量化和和除以外,是否有任何内置函数用于此操作？

Answer 1

你可以直接使用np.mean:

>>> np.mean([a, b, c], axis=0)
array([ 30.,  20.,  30.])

Answer 2

Pandas DataFrames 内置了获取列和行平均值的操作。以下代码可能对您有帮助：

import pandas and numpy
import pandas as pd
import numpy as np

# Define a DataFrame
df = pd.DataFrame([
np.arange(1,5), 
np.arange(6,10),
np.arange(11,15)
])

# Get column means by adding the '.mean' argument
# to the name of your pandas Data Frame
# and specifying the axis

column_means = df.mean(axis = 0)

'''
print(column_means)

0    6.0
1    7.0
2    8.0
3    9.0
dtype: float64
'''   

# Get row means by adding the '.mean' argument
# to the name of your pandas Data Frame
# and specifying the axis

row_means = df.mean(axis = 1)
'''
print(row_means)

0     2.5
1     7.5
2    12.5
dtype: float64
'''