从 x_arr 条件 y_arr 中搜索并将 n 个元素推入 custom_arr

Search and Push n elements into custom_arr from x_arr where condition with y_arr

抱歉,如果标题有点混乱,我不知道我还能如何让这个问题更具体。

我正在尝试在 Python 中创建 Adaboost 实现,我正在使用来自 Keras 数据集的 MNIST。

目前,我正在尝试为对“0”号图像进行分类的弱阈值创建一个训练数组。

为此,我需要创建一个数组,其中一半只是“0”的图像,另一半是任何其他随机数。

目前,我有 2 个数组,x_train:一个包含图片的数组和 y_train:一个包含标签的数组,这样我们可以检查是否,例如,x_train[i]是数字“0”的图片if y_train[i] == 0.

所以,我想知道是否有使用 NumPy 的自动化方法,使用应用于另一个数组的条件从数组中获取元素。

基本上:如果 y_array[i] == 0 ,则从 x_array[i] 抓取 n 个元素并推入 custom_array ,然后抓取 n 个元素并推入 custom_array 来自 x_array[i] 如果 y_array[i] != 0.

此致。

这是否符合您的目的?

mask = y_array == 0

array_0 = x_array[mask]
array_non0 = x_array[~mask]

如果我对你的问题的理解有误,请指正。