numpy - 通过阈值向量进行矩阵阈值化

Question

我有一个形状为 [N,M] 的 numpy 2D 数组和一个形状为 M 的 thresholds 1D 数组。如果索引 (X,i) 中的值（其中X 可以是任意行，X 介于 0 和 M-1 之间）低于阈值向量中的值，在索引 i 中。

非循环或列表理解解决方案将不胜感激。

标签： numpymatrixvectorthreshold

Answer 1

这是您可以使用的片段...

data=np.random.rand(5,3)
threshold=np.random.rand(3)

data[data-threshold<0]=-np.Inf

我使用随机数据进行测试，如果您的数据已经存在，则不需要前两行。