如何通过创建单独的二进制数组来高效地分割NumPy数组

Question

我试图在以NumPy数组形式存储的图像中分离不同的对象。例如，如果我有2D NumPy数组：

X = np.array([[1.4E14, 1.4E14, 1.6E14], [1.4E14, 1.6E14, 1.6E14], [1.6E14, 1.1E14, 1.1E14]])

如何通过实例将其拆分为相同形状的二进制数组，如下所示：

Y = np.array([[1, 1, 0], [1, 0, 0], [0, 0, 0]])
Z = np.array([[0, 0, 1], [0, 1, 1], [1, 0, 0]])
W = np.array([[0, 0, 0], [0, 0, 0], [0, 1, 1]])

其中Y表示1.4E14的值，Z表示1.6E14的值，W表示1.1E14的值，所有实例的形状与原始数组相同。对于如何在大数组中有效地执行此操作，有什么想法吗？你知道吗

Answer 1

您可以使用NumPy的^{}函数在X中查找唯一的值，^{}将X元素与这些值进行比较，^{}方法将结果转换为整数：

In [76]: import numpy as np

In [77]: X = np.array([[1.4E14, 1.4E14, 1.6E14], 
    ...:               [1.4E14, 1.6E14, 1.6E14], 
    ...:               [1.6E14, 1.1E14, 1.1E14]])
    ...: 

In [78]: for value in np.unique(X):
    ...:     binary_array = np.equal(X, value).astype(int)
    ...:     print(f'{value:.1E}\n{binary_array}\n')
    ...: 
    ...: 
1.1E+14
[[0 0 0]
 [0 0 0]
 [0 1 1]]

1.4E+14
[[1 1 0]
 [1 0 0]
 [0 0 0]]

1.6E+14
[[0 0 1]
 [0 1 1]
 [1 0 0]]

在上面的代码中binary_array = np.int_(X == value)也可以工作。你知道吗