import numpy as np
# 向量的布尔索引
x_1 = np.array([10,20,3,40])
# 输出value>15的元素
print(x_1[x_1 > 15])
# [20 40]
# 矩阵的布尔索引
x_2 = np.array([[10,20,30],[50,40,10],[10,1,10]])
print(x_2[x_2>25])
# [30 50 40]
# 找出第二列大于10的行向量
print(x_2[x_2[:,1]>10])
# [[10 20 30] [50 40 10]]
Numpy之布尔索引
最新推荐文章于 2024-08-04 09:29:52 发布