在一维NumPy数组中翻转零和一

Question

我有一个由零组成的一维NumPy数组,如下所示:

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

我想快速"翻转"这些值,使得零变为1,并且变为零,从而产生如下的NumPy数组:

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

这有一个简单的单行程吗？我查看了该fliplr()函数,但这似乎需要NumPy维数为2或更大的数组.我确信这是一个相当简单的答案,但任何帮助都会受到赞赏.

Answer 1

你的Q中肯定有一些我不明白的东西......

无论如何

In [2]: from numpy import array

In [3]: a = array((1,0,0,1,1,0,0))

In [4]: b = 1-a

In [5]: print a ; print b
[1 0 0 1 1 0 0]
[0 1 1 0 0 1 1]

In [6]: 

Answer 2

您可能应该使用布尔数据类型的标志

a = np.array([0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], dtype=np.bool)
# or
b = ~a
b = np.logical_not(a)

Answer 3

另一个多余的选择：

numpy.logical_not(a).astype(int)