python - 根据Python中其他数组中值的条件逻辑创建数组

Question

我有一个 np 数组(系列？)，数字范围从 1 到 100，我有第二个数组和一些随机数。两个数组的长度相同。我想创建一个第三个数组，如果 array2[i] > array1[i]，则为 1；如果 array2[i] = array1[i]，则为 0；如果 array2[i] < array1[i]，则为 -1。

我可以使用 for 循环和一些“if”语句轻松完成此操作，但处理时间超出了我的承受能力。

请原谅我写这篇文章的方式，我是 python 的新手，在 stackoverflow 中提问。非常感谢您的帮助。

array1 = [1,2,3,4,5,6,7,8,9,10]
array2 = [5,8,2,4,9,3,0,2,8,5]
array3 = [1,1,-1,0,1,-1,-1,-1,-1,-1]

Answer 1

这似乎是 np.sign 的一个很好的用例，它将正数变成 1，零变成 0，负数变成 -1:

>>> array1 = np.array([1,2,3,4,5,6,7,8,9,10])
>>> array2 = np.array([5,8,2,4,9,3,0,2,8,5])
>>> np.sign(array2-array1)
array([ 1,  1, -1,  0,  1, -1, -1, -1, -1, -1])