python - 有效地减少一个大的 np.array

Question

我需要减少类型的非零数组：

a = np.zeros([10**4,10**4])

以类型的简单约束为条件的较少数量的维度（列）：

a[column_index].sum() > threshold

我可以通过迭代和调用 .delete 轻松做到这一点，但我实际上正在寻找更有效的东西（因为数组很大）。有什么提示吗？

Answer 1

沿轴 0 求和以获得所有列的总和，然后创建一个布尔数组来选择列：

import numpy as np

a = np.random.randint(0, 100, [100,100])
b = a[:, a.sum(axis=0) > 5000]
print b.sum(axis=0)

输出：

array([5359, 5045, 5116, 5512, 5143, 5261, 5209, 5018, 5009, 5025, 5353,
       5149, 5407, 5258, 5148, 5527, 5176, 5173, 5028, 5110, 5406, 5211,
       5287, 5163, 5364, 5623, 5257, 5361, 5528, 5049, 5298, 5280, 5201,
       5099, 5314, 5071, 5318, 5076, 5005, 5032, 5194, 5411, 5329, 5293])