python - 如何根据相对值对 numpy 数组的每个元素进行操作？

Question

假设我们有一个数组

a = np.array([10,30,50, 20, 10, 90, 0, 25])

我想要的伪代码-

if a[x] > 80 then perform funcA on a[x]
if 40 < a[x] <= 80 then perform funcB on a[x]
if a[x] <= 40 then perform funcC on a[x]

使用 numpy 函数执行此操作的最简洁方法是什么？

Answer 1

通常，您会尝试避免对 NumPy 数组进行任何 Python 循环——这就是您首先使用 NumPy 的原因。为了举例，我假设 funcA() 将所有元素加 1，funcB() 加 2，funcC() 加 3 (请详细说明他们真正做了什么，以获得更量身定制的示例)。要实现你想要的，你可以这样做:

subset_a = a > 80
subset_b = (40 < a) & (a <= 80)
subset_c = a <= 40
a[subset_a] += 1
a[subset_b] += 2
a[subset_c] += 3

这使用 NumPy advanced indexing .例如，a > 80 的计算结果为 bool 值数组，可用于选择数组中满足条件的条目。