python - 对两个 NumPy 数组中的正元素和负元素求和

Question

>>> x1
array([[ 0.,  -1.,   2.],
       [ 3.,  -4.,  2.],
       [ -2.,  1.,  -8.]])

>>> x3
array([[ 0.,  -5.,  2.],
       [ 3.,  0.,  -3.],
       [ 3.,  2.,  8.]])

我需要输出两个矩阵:S和T，这样X是中所有正值的总和X和Y，T是X和Y中所有负值的总和。

例如:

      S = array([   [ 0.,  0.,  4.],
                    [ 6.,  0.,  2.],
                    [ 3.,  3.,  8.]])

      T = array([   [ 0.,  -6.,  0.],
                    [ 0.,  -4.,  -3.],
                    [ -2.,  0.,  -8.]])  

我使用的是 Python 2.6.7。

Answer 1

您可以使用np.clip()有选择地添加

In [140]: x1.clip(min=0) + x3.clip(min=0)
Out[140]:
array([[ 0.,  0.,  4.],
       [ 6.,  0.,  2.],
       [ 3.,  3.,  8.]])

In [141]: x1.clip(max=0) + x3.clip(max=0)
Out[141]:
array([[ 0., -6.,  0.],
       [ 0., -4., -3.],
       [-2.,  0., -8.]])