如何安排循环以遍历列然后执行某些操作

Question

我是 python 的新手，我目前正在尝试解决一个问题，该问题允许我取每列的平均值，但列数未知。

如果我知道它有多少列并单独进行每个计算，我想怎么做。我应该通过创建一个空列表并将列循环回其中来做到这一点。

import numpy as np
#average of all data not including NAN
def average (dataset):
    return np.mean (dataset [np.isfinite (dataset)])
#this is how I did it by each column separate
dataset = np.genfromtxt("some file")
print (average(dataset [:,0]))    
print (average(dataset [:,1]))
#what I'm trying to do with a loop
def avg (dataset): 
    for column in dataset:
        lst = []
        column = #i'm not sure how to define how many columns I have
        Avg = average (column)
        return Avg

Answer 1

你可以使用 numpy.mean() 函数：

https://docs.scipy.org/doc/numpy/reference/generated/numpy.mean.html

与：

np.mean(my_data, axis=0)

轴表示您是沿列还是沿行取平均值（轴 = 0 表示您取每列的平均值，即您要执行的操作）。输出将是一个向量，其长度与取平均值的列（或行）数相同，每个元素是相应列（或行）的平均值。您无需事先知道矩阵的形状即可。

您可以使用 for 循环来执行此操作，但这不是一个好主意 - 在 numpy 中循环矩阵很慢，而使用像 np.mean() 这样的向量化操作非常快。因此，一般来说，当使用 numpy 时，会尝试使用这些类型的内置操作，而不是尽可能循环遍历所有内容。

另外——如果你想要矩阵中的列数——它是

my_matrix.shape[1]

返回列数；

my_matrix.shape[0] 是行数。