python数组去掉某些元素_python – 使用条件检查从numpy数组中删除某些元素

我假设你的意思是< -100或a> -100,最简洁的方法是使用逻辑索引.

a = a[(a >= -100) & (a <= 100)]

这并不是完全“删除”条目,而是将数组的副本减去不需要的值并将其分配给先前分配给旧数组的变量.发生这种情况后,旧数组没有剩余的引用并被垃圾收集,这意味着它的内存被释放.

值得注意的是,这种方法不使用常量内存,因为我们制作数组的副本,它使用数组大小​​的内存线性.如果您的阵列太大而无法达到计算机内存的限制,这可能会很糟糕.实际通过并“移除”数组中的每个元素(即使用常量内存)的过程将是一个非常不同的操作,因为数组中的元素需要被交换并且内存块被调整大小.我不确定你是否可以使用numpy数组执行此操作,但是您可以做的一件事是避免复制是使用numpy蒙片数组：

import numpy.ma as ma

mx = ma.masked_array(a, mask = ((a < -100) | (a > 100)) )

掩码数组上的所有操作都将表现为我们“删除”的元素不存在,但我们并没有真正“删除”它们,它们仍然存在于内存中,只有一个记录显示要跳过哪些元素与数组相关联,我们不需要在内存中复制数组.此外,如果我们想要删除我们删除的值,我们可以像这样删除掩码：

mx.mask = ma.nomask