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

Question

我有一个形状为 [N,M] 的 numpy 二维数组和一个形状为 M 的阈值一维数组。 我想为 2D 矩阵中的每个元素设置负无穷大值，如果索引 (X,i) 中的值（其中 X 可以是任何行，X 介于 0 和 M-1 之间）低于阈值向量，在索引 i 中。

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

Answer 1

这是一个可以使用的 sn-p...

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

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

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