如何检测numpy数组中元素的符号变化

Question

我有一个带有正负值的numpy数组。

a = array([1,1,-1,-2,-3,4,5])

我想创建另一个数组，该数组在发生索引变化的每个索引处包含一个值（例如，如果当前元素为正，而前一个元素为负，反之亦然）。

对于上面的数组，我希望得到以下结果

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

或者，可以使用数组中发生符号变化的位置列表或布尔值列表（而不是0和1）代替。

Answer 1

就像是

a = array([1,1,-1,-2,-3,4,5])
asign = np.sign(a)
signchange = ((np.roll(asign, 1) - asign) != 0).astype(int)
print signchange
array([0, 0, 1, 0, 0, 1, 0])

现在，numpy.roll进行循环移位，因此，如果最后一个元素的符号与第一个元素不同，则signchange数组中的第一个元素将为1。如果不需要，当然可以做一个简单的操作

signchange[0] = 0

另外，np.sign认为0具有其自己的符号，不同于正值或负值。例如，即使零线仅“交叉”一次，[-1,0,1]的“
signchange”数组也将为[0,1,1]。如果不希望这样，可以插入行

sz = asign == 0
while sz.any():
    asign[sz] = np.roll(asign, 1)[sz]
    sz = asign == 0

在第一个示例中的第2行和第3行之间。