返回包含每个指定列的平均值的数组

Question

为了找到出路我都快疯了。你知道吗

输入一个二维数组和若干列数字，以数组的形式返回这些特定列中每个数字的平均值。你知道吗

我知道如何输出所有列的平均值，但不知道如何只输出特定列的平均值。你知道吗

a=数组（[[0，0,1]，[1，1,2]，[3，3,3]]）

获取平均值（a，[0,1]）

数组（[1.333，1.333]）

Answer 1

在numpy

中求平均值

numpy有一个average函数：

>>> import numpy as np
>>> a = np.array([[0, 0,1], [1, 1,2], [3, 3,3]])
>>> np.average(a[:,(0,1)], axis=0)
array([ 1.33,  1.33])

对于np.average，axis指定要平均的数组轴。axis=0，例如，行上的平均值。你知道吗

np.average还提供加权平均值，如果您需要的话。你知道吗

定义get_average函数

如果您认为仍然需要get_average函数，那么：

def get_average(a, cols):
    return np.average(a[:,cols], axis=0)