两个numpy技巧(模式识别总结)
1. 查找符合特定条件的Numpy ndarray对象中元素的数量
np.count_nonzero()
函数可以用来完成此操作。以下是使用np.count_nonzero()
函数快速查找数组中小于5的元素数量的示例:
1 |
|
在此示例中,我们使用 array < 5
来获取array
中小于5的所有元素,然后使用 np.count_nonzero()
函数来计算满足这些条件的元素的数量。该函数返回一个整数,表示满足条件的元素的数量。请注意,在此示例中,我们假设 array
是一个一维数组。如果array
是一个多维数组,则需要将其展平为一维数组,然后再使用 np.count_nonzero()
函数,或使用 axis
参数指定计数的轴。
2. 使用np.logical_not()
函数区分训练集和测试集
修改后的代码如下:
1 |
|
通过使用 np.logical_not()
函数,我们可以将train_index
取反以获取测试集数据,即除了train_set
之外的所有数据。
两个numpy技巧(模式识别总结)
https://fulequn.github.io/2023/06/Article202306061/