以下是一个示例代码,展示了如何按照向量筛选向量:
import numpy as np
# 创建测试向量
vector1 = np.array([1, 2, 3, 4, 5])
vector2 = np.array([6, 7, 8, 9, 10])
# 创建筛选条件向量
filter_vector = np.array([True, False, True, False, True])
# 使用筛选条件向量进行筛选
filtered_vector1 = vector1[filter_vector]
filtered_vector2 = vector2[filter_vector]
# 输出筛选结果
print("Filtered vector 1:", filtered_vector1)
print("Filtered vector 2:", filtered_vector2)
输出结果:
Filtered vector 1: [1 3 5]
Filtered vector 2: [ 6 8 10]
在示例代码中,首先创建了两个测试向量 vector1
和 vector2
,然后创建了一个筛选条件向量 filter_vector
,其中 True
表示对应位置的元素需要被筛选,False
表示不需要被筛选。
接下来,通过将筛选条件向量作为索引,可以实现按照条件筛选向量的功能。使用 vector[filter_vector]
的语法,可以得到筛选后的向量。
最后,将筛选结果输出到控制台,以验证结果。
上一篇:按照向量或列表进行分组或聚合