python - 如何根据 True 和 False 语句比较两个 numpy 数组

Question

self.x = np.array( [np.random.uniform( -5, 5 ) for _ in xrange( 100 )] )
v = np.array( [np.random.uniform( -5, 5 ) for _ in xrange( 100 )] )
good = np.random.uniform(0,1,5) < 0.5
good = good.reshape(1,self.x.shape[1])
self.x[good] = v[good]
u = self.x[good]

假设good是[False True True True True]。 u 向量丢弃第一个值为 False 的值，其余值为 True 的值被替换。但我希望它保留 self.x

的先前值

Answer 1

在我看来，您希望x和v保持不变，但有第三个向量u，即 v 只要 good 为 True，但 x 在其他地方呢？为此，您可以使用函数np.where。

这是一个例子:

x = np.arange(1,6)
v = 10*x
good = np.array([False, True, True, True, True])

In [690]: x
Out[690]: array([1, 2, 3, 4, 5])

In [691]: v
Out[691]: array([10, 20, 30, 40, 50])

In [692]: good
Out[692]: array([False,  True,  True,  True,  True], dtype=bool)

In [693]: u = np.where(good, v, x)

In [694]: u
Out[694]: array([ 1, 20, 30, 40, 50])

In [695]: x
Out[695]: array([1, 2, 3, 4, 5])

In [696]: v
Out[696]: array([10, 20, 30, 40, 50])