如何用Python一行代码实现保留符号的Numpy数组最小幅值提取

嘿，这事儿用Numpy的向量化操作轻松搞定一行代码，再也不用写繁琐的循环啦！

核心思路就是：对两个数组的每个元素，比较它们的绝对值大小，然后选择绝对值更小的那个元素（同时保留它原本的符号）。用np.where就能完美实现这个逻辑：

import numpy as np

# 假设你的两个等长数组是a和b
result = np.where(np.abs(a) < np.abs(b), a, b)

举个实际例子看看效果： #

a = np.array([-2, 3, -1, 5])
b = np.array([3, -1, 2, -4])

result = np.where(np.abs(a) < np.abs(b), a, b)
print(result)  # 输出: [-2, -1, -1, -4]

解释一下这个结果：

• 第一个位置：abs(-2)=2 < abs(3)=3，选-2
• 第二个位置：abs(3)=3 > abs(-1)=1，选-1
• 第三个位置：abs(-1)=1 < abs(2)=2，选-1
• 第四个位置：abs(5)=5 > abs(-4)=4，选-4

这个方法是完全向量化的，比你之前写的循环效率高太多——尤其是当数组规模很大的时候，Numpy的底层C实现会让运算速度提升一个量级。

内容的提问来源于stack exchange，提问作者ClimateUnboxed