比较列表/子列表

Question

我有2个清单：

lista = ['1.2.3.4', '2.3.4.5', '3.4.5.6'] # 12,000 IP's
listb = [['1.2.3.4', 'info', 'moreinfo', 'moremoreinfo'], ['2.3.4.5', 'info', 'moreinfo', 'moremoreinfo']] # 1.8m IP's + info

我正在寻找一种获取lista的方法，如果ip在listb中存在，请给我所有有关它的信息。

我已经尝试过循环，但是它的速度非常慢：

for listaitem in lista:
    for listbitem in listb:
        if listaitem in listbitem[0]:
            print listbitem

关于如何加快速度的任何建议？

Answer 1

您可以将lista变成一组以进行快速成员资格测试，然后循环遍历listb以选择在lista中找到的任何一个：

lista_set = set(lista)
for item in listb:
    if item[0] in lista_set:
        print item

下一步是将listb变成字典：

listb_dict = {item[0]: item[1:] for item in listb}

现在，您可以使用集合来仅选择lista_set和listb_dict ：

for match in listb_dict.viewkeys() & lista_set:
    print match, listb_dict[match]

Answer 2

lista = ['1.2.3.4', '2.3.4.5', '3.4.5.6']
listb = [['1.2.3.4', 'info', 'moreinfo', 'moremoreinfo'],
         ['2.3.4.5', 'info', 'moreinfo', 'moremoreinfo']]

将listb变成字典

dictb = {i[0] : i[1:] for i in listb}

遍历lista并在dictb查找条目

for elem in lista:
    print dictb.get(elem)

['info', 'moreinfo', 'moremoreinfo']
['info', 'moreinfo', 'moremoreinfo']
None

Answer 3

您应该将数据转换为更适合搜索的格式：字典。

ip_info = {info[0]: info[1:] for info in listb}

然后，您可以非常快速地查找有关特定IP的信息。

for ip in lista:
    if ip in ip_info:
        print(ip_info[ip])