使用位数组创建类似对象的列表

Question

我需要在 Python 中跟踪一组可能有 1000 万个数字。 （所有数字都在 0 到 2^32 之间）。我会事先知道整数的最大值，并且在 0 和最大值之间，20-80% 的值将在集合中。

我当前的代码使用内置的set。这种方式太慢了。就性能而言，最好的方法是使用位数组（例如 https://pypi.python.org/pypi/bitarray/ ）。

我很容易使用位数组来构建具有add(n) 和remove(n) 方法的类。我不知道该怎么做是支持for n in bitarray_set:。我想我需要使用迭代器或可迭代的，但我不知道该怎么做。这可能吗？怎么样？

Answer 1

bitarrays 支持itersearch 方法，该方法迭代一个位数组出现在另一个位数组的所有位置。使用它：

def __iter__(self):
    return self.bits.itersearch(bitarray([True]))