将numpy数组转换为以0为正的符号数组

Question

我有一个大的numpy数组，带有正数据，负数据和0。 我想将其转换为具有当前值符号的数组，以便将0视为正数。 如果我使用numpy.sign则如果当前值为0，则它​​返回0，但是我想要返回1的东西。 是否有捷径可寻？

Answer 1

如果x是数组，则可以使用2*(x >= 0) - 1 。

x >= 0将是一个布尔值数组（即False和True ），但是当您对其进行算术运算时，它将有效地转换为0和1的数组。

你也可以做np.sign(x) + (x == 0) 。 （请注意，即使x是整数数组， np.sign(x)返回浮点值。）