使用列表理解来搜索一个二维数组（python）

Question

我想搜索一个二维数组的字符并返回数组中所有字母T的数组索引（x_T，y_T）。我想这可以很容易地用两个堆叠的for循环完成，但我很想知道是否可以使用列表理解来有效地完成。 我已经试过了诸如：

TPos = [[x, y] for x in range(0, len(array[x])) for y in range(0, len(array[x][y])) if array[x][y] == 'T'] 

但我得到错误做数组边界。有人能指出我的方向是正确的吗？欢呼声中， 杰克

编辑

现在试图用“ndenumerate”，像这样：

TPos = [pos for pos, x in numpy.ndenumerate(array) if x == "T"] 

Answer 1

这最终是一个野兽一行，为便于阅读，可能更容易打破它向上。

l = [['a','b','T'], 
    ['T','T','g'], 
    ['c','T','T']] 

reduce(lambda x,y: x+y, [[(x,y) for y in range(len(l[y])) if l[x][y] == 'T'] for x in range(len(l))]) 

输出

[(0, 2), (1, 0), (1, 1), (2, 1), (2, 2)] 

编辑随着numpy这要简单得多。

import numpy as np 
[pos for pos, x in np.ndenumerate(np.array(l)) if x == 'T'] 

输出

[(0, 2), (1, 0), (1, 1), (2, 1), (2, 2)] 

Answer 2

a=[[1,2,3,4],[5,6,7,8]]

def return_indexes(some_list,what_you_want_to_find): 
    temp=[] 
    for i in range(len(some_list)): 
     for j in range(len(a[i])): 
      if a[i][j] == what_you_want_to_find: 
       temp.append([i,j]) 
    return temp 

print return_indexes(a,7) 

Answer 3

不太一样好@网络的答案（即，元组列表而不是元组的名单列表），但使用“香草”列表内涵与枚举（嘿，我很好奇）：

l = [['a','b','T'],['T','T','g'],['c','T','T']] 
[[(i,j) for j,col in enumerate(row) if col=='T'] for i,row in enumerate(l)] 

输出

[[(0, 2)], [(1, 0), (1, 1)], [(2, 1), (2, 2)]] 

当然

你可以申请reduce(lambda ...)或similar approach此得到一个元组列表。

Answer 4

numpy的where这是否只是：

In [16]: l = [['a','b','T'], 
    ....:  ['T','T','g'], 
    ....:  ['c','T','T']] 

In [20]: np.where(np.array(l)=='T') 
Out[20]: (array([0, 1, 1, 2, 2]), array([2, 0, 1, 1, 2])) 

In [22]: zip(*np.where(np.array(l)=='T')) 
Out[22]: [(0, 2), (1, 0), (1, 1), (2, 1), (2, 2)] 

没有numpy这里是我的版本的列表理解的：

In [27]: [(i,j) for i,v in enumerate(l) for j,k in enumerate(v) if k=='T'] 
Out[27]: [(0, 2), (1, 0), (1, 1), (2, 1), (2, 2)] 

Answer 5

问的问题基本上是试图实现这种

array= [['a','b','T'], ['T','T','g'], ['c','T','T']] 
TPos = [(x, y) for x in range(len(array)) for y in range(len(array[x])) if array[x][y] == 'T'] 
TPos 

输出

[(0, 2), (1, 0), (1, 1), (2, 1), (2, 2)]