Python:在数组列中查找特定值

Question

我有一个特定的整数值 k，我想在二维 numpy 数组的列中找到匹配值(使用“for”循环)。我想我可以使用 if 语句并直接将单个数组元素与整数值 k 进行比较。这是我的代码:

for i in range(mock.shape[0]):

    n_cl = int(mock[i,0]/3500.)
    zcl = mock[i,5]
    pick = [np.random.random_integers(200, size=(n_cl))]
    print pick[0]   
    if(zcl <= 0.05):

        for k in range(len(pick)) :

            for j in range(z_001.shape[0]):
                n = z_001[j,1]
                if (int(n) == pick[k]):
                    binaries[j,7] = mock[i,0]   
                    binaries[j,8] = mock[i,1]
                    binaries[j,9] = mock[i,2]
                    binaries[j,10] = mock[i,3]
                    binaries[j,11] = mock[i,4]

我总是得到一个关于具有多个元素的数组的真值的 ValueError，这是不明确的。我知道问题出在“int(n) == k”，但是我想知道我哪里错了，我怎样才能把事情做得更好，让我的这部分代码工作。

Answer 1

好吧，考虑一下:

1. 无论何时使用 numpy，您可能都不想使用 for 循环
2. 您得到的错误意味着您正在将 numpy 数组与单个值进行比较

因此，该问题的一种可能解决方案是:

import numpy as np

# generating some fake data
x = np.repeat([[1,2,3,4]], 4, axis=0) 
x[2,2] = 200
x[3,1] = 200

# retrieving the indices where "x==200" using np.where
indices = np.where(x==200)
print indices

给出:

(array([2, 3]), array([2, 1]))

因此您可以使用 indices 索引任何其他数组(具有适当的形状)您将在 indices 位置获得该数组的值。