python - 根据逻辑条件更改一些numpy数组列

Question

我有一个2D NumPy 数组，我想执行以下操作:

对于数组中的每一列，即一系列非递减值，将此列替换为差异列(即，每个条目都是前两个条目之间的差异)。

其他每一列都保持不变(除了第一行被删除以适应差异列维度)。

例如矩阵中:

[ [1,1,1,2,3,4]
  [1,3,4,3,4,5]
  [1,7,3,4,2,7] ]

差异矩阵是:

[ [0,2,3,1,1,1]
  [0,4,-1,1,-1,2] ]

因此，具有递减值的第三列和第五列将保持不变，而其他列将替换为差异列，结果为:

[ [0,2,4,1,4,1]
  [0,4,3,1,2,2] ]

我试过这样的:

tempX = np.diff(X, axis = 0).transpose()
return np.where(tempX >= 0, tempX, X[1:].transpose())

但是 np.where 中的条件是按元素执行的，而不是针对每一列(或行)执行的。

我怎样才能改变条件让它起作用？有没有更有效的方法来实现这一点？

Answer 1

你可以这样试试:

b = a[1:] - a[:-1]
decrease = numpy.where(numpy.min(b, axis=0)<0)
b[:,decrease] = a[1:, decrease]

你也可以在一个表达式中做到这一点:

numpy.where(numpy.min(a[1:]-a[:-1],axis=0)>=0, a[1:]-a[:-1], a[1:])