python中的二维数组索引值

Question

这应该不会那么难。假设我有一个二维数组：

a = [['1' 'George'] 
    ['5' ' '] 
    ['7' 'Jose'] 
    ['5' ' '] 
    ['7','Fred']] 

我希望能够找到所有的索引值，其中a[:,1] == ' '

我最好的猜测是：

missing_vals = a[a[:,' ']==' ' 
a[missing_vals] 

我不想要的答案：

['5','5'] 

但答案：

[1,4] 

意味着数组的第2和第5个元素。

谢谢。

Answer 1

这是你在找什么：

>>> a = [['1', 'George'], 
...  ['5', ' '], 
...  ['7', 'Jose'], 
...  ['5', ' '], 
...  ['7','Fred']] 
>>> [i for i, [k,v] in enumerate(a) if v == ' '] 
[1, 3] 

说明：

我们在清单a在这a元素[k,v]有元素v等于空间要求所有索引i 。