python - 在 Python 中查找矩阵中某个元素的所有出现位置

Question

我有一个列表列表，我想找到所有事件的坐标。我设法做到了，但我想知道是否有更好的方法使用 numpy where 例如。

这就是我所做的:

my_list = [[1,2,3,1, 3], [1,3,2]]

target_value = 3
locations = []
for k in range(len(my_list)):
    indices = [i for i, x in enumerate(my_list[k]) if x == target_value]
    locations.append((k, indices))
locations2 = []
for row in locations:
    for i in row[1]:
        locations2.append((row[0], i))
print locations2 # prints [(0, 2), (0, 4), (1, 1)]

Answer 1

虽然你可以让它在 numpy 中工作，但 numpy 对参差不齐的数组并不那么满意。我认为纯Python理解版本看起来还不错:

>>> my_list = [[1,2,3,1, 3], [1,3,2]]
>>> [(i,j) for i,x in enumerate(my_list) for j,y in enumerate(x) if y == 3]
[(0, 2), (0, 4), (1, 1)]