查看是否有特定值存在,及特定值所在行或者列,并返回布尔类型的一个矩阵
一、① 假设 t2 为一个二维的 3*3 的矩阵,查看是否有等于25的值
#打印t2
t2 = np.array([[5,10,15],[20,25,30],[35,40,45]])
print(t2)
>>[[ 5 10 15]
[20 25 30]
[35 40 45]]
② 查看t2的形状
print(t2.shape)
>>(3, 3)
③ 查看t2中是否有等于25的值存在,并输出结果
t2 == 25
>>array([[False, False, False],
[False, True, False],
[False, False, False]], dtype=bool)
二、接下来,我们将返回的布尔类型矩阵作为索引,会得到什么呢?
a = t2 == 25
print(t2[a])
>>[25]
三、要是查找整列是否有特定值存在呢?查看第二列是否有等于25的值,如下图:
b = (t2[:,1] == 25)
print(b)
>>[False True False]
四、或者查找特定值所在的行、列的数据,又该如何找呢?查找出25所在行的数据并打印结果,如下图:
print(t2[b,:])
>>[[20 25 30]]