首页 > 解决方案 > 屏蔽numpy数组中的列

问题描述

从 numpy 数组中屏蔽列的最有效方法是什么?

a = np.arange(12).reshape((3,4))
mask = [True False True False]

我正在寻找类似的东西:

b = a[mask, axis=1]

这将给出输出:

b = [[0, 2],[4, 6], [8, 10]]

我需要这个用于大型阵列。

谢谢,

标签: arraysnumpymask

解决方案


推荐阅读