首页 > 解决方案 > 附加一个数组中的值,如果另一个数组中的值大于某个值,则附加该值

问题描述

例如说我有两个长度相等的数组。

x = np.array([-34, 1, -12, 4, 5, -5])

y = np.array([2, -1, 4, 6, 7, 4])

我想创建一个新的数组或相等长度的列表,如果 x 的值小于某个数字 (g),那么它会附加 (g),否则它将附加 y。

因此,在上面的两个数组中,如果 x 小于 -5,它将追加 -5,否则将追加 y 的值。结果会是这样:

new_array = ([-5, -1, -5, 6, 7, 4])

标签: pythonarraysappendcomparison

解决方案


您可以使用 简单地执行此操作np.where,例如:

In []:
np.where(x < -5, -5, y)

Out[]:
array([-5, -1, -5,  6,  7,  4])

推荐阅读