比numpy更快吗？

Question

我想到的是否比NP的功能更快。 ATM我正在将两个数组与NP进行比较。之后，IM浏览列表检查值[a]的位置。我认为必须有一种更快的方法，但我目前找不到。对数组进行排序是没有选项。

代码看起来像这样：

a = np.array([1, 2, 3])
b = np.array([2, 1, 8])
over = np.less(a, b)
# over = [True, False, True]
pos = np.where(over == True)[0]
# pos = [0, 2]
x = b[pos]
# x = [2,8]

Answer 1

AS ali_sh指出，只需使用条件来获取值即可。

import numpy as np

a = np.array([1, 2, 3])
b = np.array([2, 1, 8])

b[np.less(a, b)]
# array([2, 8])
list(b[np.less(a, b)])  # for a list result instead of np arr
# [2, 8]

# or just
b[a < b]
# array([2, 8])

问题不是np.less的速度，而是您获得的额外步骤。所有这些几乎是速度快速的；只需要删除不必要的步骤即可。

即使您想在上保存在中的true/false结果：

over = np.less(a, b)
b[over]
# array([2, 8])

over = a < b
b[over]
# array([2, 8])