Numpy.where與邏輯

Question

000000000
000050000
000505000
005000500
050000050
500040005
050000050
005000500

在此示例中，我想找到與4處於同一行的5s的索引。我已經將4的索引保存為centre 。

所需的輸出是[[5,0],[5,8]] 。

我嘗試了此操作，但不起作用：

index = zip(*np.where(array== 5 and x == centre[0]))

Answer 1

列表理解的解決方案：

arr = ['000000000',
       '000050000',
       '000505000',
       '005000500',
       '050000050',
       '500040005',
       '050000050',
       '005000500']

indices = [[j, i] 
           for j, row in enumerate(arr)
           for i, val in enumerate(row) 
           if '4' in row and val=='5']

print(indices)
# [[5, 0], [5, 8]]

Answer 2

通過執行index = zip（* np.where（array [centre [0] ,:] == 5）來解決它