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

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() 。