有效地减少大的np.array

Question

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

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

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

a[column_index].sum() > threshold

我可以通过迭代和调用.delete来轻松完成它，但我实际上正在寻找更高效的东西（因为数组非常大）。任何提示？

Answer 1

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

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])