Python | 多维矩阵如何读取满足判定条件的数据

引言

矩阵的读取与数据挑选其实并不是一个复杂的问题，但是当矩阵为多维矩阵时，编写“按照判定条件保留矩阵中符合条件的数据，将不符合条件的数据设置为0”这样一个问题时，却遇到了许多的问题，导致结果与我想要的不符合。

总结

1. 初始代码

一般情况下，针对矩阵或者数组，采用Ture or False读取数据即可。
示例代码：

A = np.array([1,2,3,4,5,6,7,8,9]).reshape((3, 3))
ind = A ==5  # 判定条件
result = A[ind]  # 读取符合判定条件的矩阵的数据

输出结果为： result = [5], 说明两点：

1. 输出矩阵都改变了数组本来的结构，原矩阵为3*3数组，输出却为1维数组
2. 输出的结果中，并没有保留不满足判定条件的数据的位置并将其设置为0，不符合我们的要求。

2. 改进

换一种思路想，其实要想实现读取满足判定条件的数据，只需要保留原矩阵，并将不满足判定条件的数据设置为0即可。

因此，代码改为：

data_new = data.copy()
data_new [data_new != 5] = 0  # 读取分类后的图像像素LAB值：类别为2

参考网址

1. https://blog.csdn.net/u013044310/article/details/85948958
2. https://blog.csdn.net/u012530451/article/details/78683606/