bg算法python源码

以下是Python实现的Bloom Filter算法的基本代码示例：

import mmh3
from bitarray import bitarray

class BloomFilter:
    def __init__(self, size, hash_count):
        self.size = size
        self.hash_count = hash_count
        self.bit_array = bitarray(size)
        self.bit_array.setall(0)

    def add(self, string):
        for seed in range(self.hash_count):
            result = mmh3.hash(string, seed) % self.size
            self.bit_array[result] = 1

    def lookup(self, string):
        for seed in range(self.hash_count):
            result = mmh3.hash(string, seed) % self.size
            if self.bit_array[result] == 0:
                return "Nope"
        return "Probably"

在这个实现中，我们使用了mmh3库来实现哈希函数，bitarray库来实现位数组。在初始化时，我们需要提供两个参数，一个是位数组的大小，一个是哈希函数的数量。当我们添加一个字符串时，会使用多个哈希函数对该字符串进行哈希，并在位数组中将相应的位设置为1。当我们查找一个字符串时，同样会使用多个哈希函数对该字符串进行哈希，并检查位数组中相应的位是否都为1。如果都为1，则返回"Probably"，否则返回"Nope"。