python-3.x - 根据第二个数组的列中的值更改一个数组中的值

Question

我有一个名为 Y_train 的一维数组包含一系列 1 和 0。我有另一个名为 sample_weight 的数组这是一个全 1 的数组，形状为 Y_train ， 定义为:

sample_weight = np.ones(Y_train.shape, dtype=int)

我正在尝试更改 sample_weight 中的值到 2 ，其中对应的值在 Y_train == 0 .所以最初并排它看起来像:

Y_train        sample_weight
0              1
0              1
1              1
1              1
0              1
1              1

我希望它在转换后看起来像这样:

Y_train        sample_weight
0              2
0              2
1              1
1              1
0              2
1              1

我尝试的是使用 for循环(如下所示)，但在 sample_weight 中没有一个 1 变为 2 .我想以某种方式使用 np.where()如果可能的话，函数，但这并不重要，只是想避免 for环形:

sample_weight = np.ones(Y_train.shape, dtype=int)
for num, i in enumerate(Y_train):
    if i == 0:
        sample_weight[num] == 2

我尝试使用所示的解决方案 here但第二个阵列没有成功。有任何想法吗？？？谢谢!

Answer 1

import numpy as np

Y_train = np.array([0,0,1,1,0,1])
sample_weight = np.where(Y_train == 0, 2, Y_train)

>> print(sample_weight)
[2 2 1 1 2 1]

np.where基本上就像 Excel 的“IF”一样工作:
np.where(condition, then, else)
也适用于转置数组:

Y_train = np.array([[0,0,1,1,0,1]]).T
sample_weight = np.where(Y_train == 0, 2, Y_train)

>> print(sample_weight)
[[2]
 [2]
 [1]
 [1]
 [2]
 [1]]