将新列添加到反向二进制值的 ndarray | NumPy 数组 Python

问题描述

我想将第二列附加到 ndarray，包含相反的二进制值。

取这个 n 行 1 列的 ndarray，包含二进制值：

[[0]
 [1]
 [0]
 [1]]

我想要一个产生这个输出的 for 循环：

[[0,1]
 [1,0]
 [0,0]]

访问嵌套的 ndarray，添加相反的内容：1 或 0。

注意：append()、insert() 等不起作用，因为这是一个多维数组。不是列表（因此有很多 []）

解决方法

通过对二进制值进行 not 门操作来尝试将 0 更改为 1，反之亦然。然后可以将2个数组a和not a水平堆叠成(4,2)数组-

import numpy as np

a = np.array([[0],[1],[0],[1]])
b = np.logical_not(a).astype(int)

np.hstack([a,b])

array([[0,1],[1,0],[0,0]])

,

你可以试试：

np.concatenate([a,1-a],axis=1)

或

np.hstack([a,1-a])

输出：

array([[0,0]])