Question

我有一个'mask'（布尔）2D数组，我想把它转换成一个坐标列表。什么是适当的numpythonic方式呢？

输入将是这样的：

[[False,False,True],
 [False,True,False]]

并给出上述输入，输出应为：

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

Answer 1

使用

np.where：如果您想稍后使用它索引另一个数组，可以使用它。但结果并不完全符合您的要求。
np.argwhere：如果您需要指定的结果。但是这个结果不能用于索引另一个数组。

一些示例代码：

import numpy as np
a = np.array([[False,False,True],
              [False,True,False]])
np.argwhere(a) # equivalent to checking a == True
#array([[0, 2],
#       [1, 1]], dtype=int64)
np.where(a) # equivalent to checking a == True
#(array([0, 1], dtype=int64), array([2, 1], dtype=int64))

如果您想将结果转换为列表，则可以使用ndarray.tolist()方法。所以你可以致电np.argwhere(a).tolist()。

将mask（boolean）数组转换为x，y坐标列表

1 个答案: