如何根据二进制掩码删除numpy中的列?

时间:2018-10-05 20:14:19

标签: python numpy

我有长度为n的二进制向量。 [0,0,0,1 ....] 如何删除numpy中大小为n * m的矩阵的列,该列与二进制向量中的值1相对应?

有没有简单或优雅的方法可以做到这一点?

1 个答案:

答案 0 :(得分:2)

如果您的意思是拥有value矩阵,则可以将二进制数组用作布尔值以对矩阵进行索引:

m*n

示例:

mat[:,bin_vect.astype('bool')]