如何将Numpy数组中满足条件的元素筛选至新数组？

嘿，我来帮你搞定这个需求！你已经找对了判断元素是否符合条件的方向——x%2 == 0确实能生成一个布尔数组标记哪些元素能被2整除，接下来只要用这个布尔数组作为索引去提取原数组里的对应元素就可以得到你想要的数值啦。

方法一：Numpy布尔索引（推荐） #

这是Numpy里最简洁高效的方式，直接一步到位：

import numpy as np

x = np.array([1,2,3,4,5,6,7,8,9])
y = x[x % 2 == 0]
print(y)  # 输出: [2 4 6 8]

原理很简单：x%2 == 0会生成一个和x长度相同的布尔数组[False, True, False, True, ...]，用它去索引x时，Numpy会自动筛选出所有对应位置为True的元素，直接返回包含这些数值的新数组。

方法二：循环遍历（适合理解逻辑） #

如果你想更直观地看到筛选过程，也可以用循环逐个判断添加，最后转成Numpy数组：

import numpy as np

x = np.array([1,2,3,4,5,6,7,8,9])
y = []
for num in x:
    if num % 2 == 0:
        y.append(num)
y = np.array(y)  # 把列表转成Numpy数组
print(y)  # 输出: [2 4 6 8]

不过这种方法在处理大型数组时效率不如布尔索引，所以更推荐第一种方式哦。

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