根据系数调整numpy数组中的值[重复]

Question

我正在使用阈值向量在 2d numpy 数组中逐行创建二进制值。示例代码如下：

import numpy as np
x = np.random.rand(100000, 200)
coef = np.random.random(x.shape[1])
x = np.array([[1 if x[i,j]>=coef[j] else 0 for j in range(x.shape[1])] for i in range(x.shape[0])])

有没有办法让它更快？

Answer 1

与coef 进行比较，得到一个布尔数组，然后转换为int 数组，从而利用 NumPy 的矢量化功能 -

x_out = (x >= coef).astype(int)