Python基础：Numpy-按行/列求均值

numpy.average(a, axis=None, weights=None, returned=False)[source]

方法简介

对ndarray数组求各维度的均值。

参数说明

Parameters

• a

  array_like

  包含要平均的数据的numpy数组。如果不是数组，会尝试转换。

• axis

  {0 , 1}, default None

  若axis=0，则按照列求均值；若axis=1，则按照行求均值，若axis=None，则对整体求均值。默认axis=None

Returns

• retval, [sum_of_weights]：array_type or double

  沿指定轴返回平均值。返回True时，返回一个以平均值作为第一个元素的元组，并将权重和作为第二个元素。

用法举例

>>> import numpy as np
>>> data = np.array([[1, 2], [3, 4]])
>>> np.average(data)  # 整体求均值
2.5
>>> np.average(data, axis=0)  # 按列求均值
array([2., 3.])
>>> np.average(data, axis=1)  # 按行求均值
array([1.5, 3.5])