python - 数组中的 Numpy 条件乘法数据(如果为真乘以 A，则为假乘以 B)

Question

假设我有一个数值为 0~255 的大数组。我希望此数组中大于 100 的每个元素都乘以 1.2，否则乘以 0.8。

这听起来很简单，但除了遍历所有变量并将其一一相乘之外，我找不到其他方法。

Answer 1

如果 arr 是您的数组，那么这应该有效:

arr[arr > 100] *= 1.2
arr[arr <= 100] *= 0.8

更新:正如评论中所指出的，这可能会产生第一步影响第二步所做的事情的不良影响，因此我们应该改为做类似的事情

# first get the indexes we of the elements we want to change
gt_idx = arr > 100
le_idx = arr <= 100
# then update the array
arr[gt_idx] *= 1.2
arr[le_idx] *= 0.8