python - 根据系数调整 numpy 数组中的值

Question

这个问题已经有答案了:

Convert NumPy array to 0 or 1 based on threshold (2 个回答)

已关闭 5 年前。

我使用阈值向量在二维 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 进行比较，得到一个 bool 数组，然后转换为 int 数组，从而利用 NumPy 的矢量化功能 -

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