multidimensional-array - 迭代numpy多维数组的一列？

Question

我有一个 (2, 500) numpy 数组，名为county_data. 我想遍历第一列，检查每个值是否等于 number someNumber，如果是，则将其行附加到名为temp.

到目前为止，这是我的代码：

for entry in county_data:       
    if entry[0] == someNumber:  
        temp.append(entry)  
    print temp

这是我得到的错误：

   if entry[0] == code:
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

我不太清楚这意味着什么，a.any()anda.all()函数似乎并没有对数组中的每一行执行我想要的操作。如何编辑我的代码以检查数组每一行中的第一个条目是否匹配someNumber？

score 3 · Accepted Answer

不要那样做。相反，一次访问所有行（即，矢量化您的代码）：

temp = county_data[county_data[:, 0] == someNumber]

1 回答 1