python - 我如何在Python中循环函数变量？

Question

这是我当前的代码:

import numpy as np

vec0 = [1, 2, 3]
vec1 = [2, 3, 4]
vec2 = [3, 4, 5]
vec3 = [4, 5, 6]

for k in range(0, 4):
    globals()['mean%s' % k] = np.mean('vec'+str(k))

我收到此错误:

类型错误:无法使用灵活类型执行reduce

<小时/>

我想要这个结果。

mean0 = np.mean(vec0)
mean1 = np.mean(vec1)
mean2 = np.mean(vec2)
mean3 = np.mean(vec3)

Answer 1

按名称调用变量是一种反模式。如果您需要对多个对象执行任务，您可以构造这些对象的集合(元组、列表等)。例如:

all_vecs = [vec0, vec1, vec2, vec3]

此外，您现在可以通过指定 axis 参数，使用 numpy 轻松处理批量平均值:

all_means = np.mean(all_vecs,axis=1)

然后:

>>> all_means
array([ 2.,  3.,  4.,  5.])