python - 将输出转换为 numpy 数组

Question

我面临以下问题。我有 4 个数据文件 (Data_1...Data_4)，我试图获取第 2 列中小于或等于 5 的项目数。我的以下代码可以完成这项工作。

import numpy as np
filelist=[]
for i in list(range(1,5)):
    filelist.append("/Users/Hrihaan/Desktop/Data_%s.txt" %i)
for fname in filelist:
    data=np.loadtxt(fname)
    z=data[:,1]
    count= len([i for i in z if i <= -5]) # output 5 3 0 9
    x= np.array(count)
    Average=np.mean(x)
    print(Average)

但是我被困在如何处理输出(5,3,0,9)上，我想从输出(计数)创建一个数组来进行简单的数学计算，例如查找平均值或中位数。所以平均值应该是 (5+3+0+9/4 = 4.25)，但是当我尝试打印(平均值)时，我得到相同的输出 5 3 0 9，而不是平均值。

任何帮助将不胜感激。

Answer 1

不要为循环中的每个循环创建一个数组，而是首先创建一个列表，并在循环完成后将此列表转换为数组。

import numpy as np
filelist=[]
for i in list(range(1,5)):
    filelist.append("/Users/Hrihaan/Desktop/Data_%s.txt" %i)

counts = []
for fname in filelist:
    data=np.loadtxt(fname)
    z=data[:,1]
    count= len([i for i in z if i <= -5]) # output 5 3 0 9
    counts.append(count)

x = np.array(counts)
Average=np.mean(x)
print(Average)