Question

我必须找到有效的python代码才能执行以下操作：

找到两个给定列表中包含的n个连续元素的序列，如果有的话，至少一个。

例如，对于n=3，这两个列表的结果将为['Tom', 'Sam', 'Jill']：

lst1 = ['John', 'Jim', 'Tom', 'Sam', 'Jill', 'Chris']
lst2 = ['Chris', 'John', 'Tom', 'Sam', 'Jill', 'Jim']

下面的示例代码可以达到目的，但是如果我必须比较成千上万的行/列表，那么它将永远需要做同样的事情。对于如何优化此代码的性能以处理大量数据的任何建议，将不胜感激！

lst1 = ['John', 'Jim', 'Tom', 'Sam', 'Jill', 'Chris']
lst2 = ['Chris', 'John', 'Tom', 'Sam', 'Jill', 'Jim']
strNum = 3 #represents number of consecutive strings to search for
common_element_found = 'False'
common_elements = []

lst1length = len(lst1) - (strNum - 1)
lst2length = len(lst2) - (strNum - 1)
for x in range(lst1length):
    ConsecutiveStringX = lst1[x] + ' ' + lst1[x + 1] + ' ' + lst1[x + 2]
    for y in range(lst2length):
        ConsecutiveStringY = lst2[y] + ' ' + lst2[y + 1] + ' ' + lst2[y + 2]
        if ConsecutiveStringY == ConsecutiveStringX:
            common_element_found = 'True'
            common_elements.append(ConsecutiveStringY)
            print('Match found: ' + str(common_elements)) 
            break
    if common_element_found == 'True':
        common_element_found = 'False'
        break

Answer 1

IIUC，

consecs1 = [ tuple(lst1[i:i+3]) for i in range(0, len(lst1)-2)]
consecs2 = { tuple(lst2[i:i+3]) for i in range(0, len(lst2)-2)}

for c in consecs1: 
    if c in consecs2:
         print(c)

输出

('Tom', 'Sam', 'Jill')

说明：您可以为tuples列出lst1的列表，它们是可哈希对象，并检查它们是否是in的{{1}}的{{1}}由set构建（授予O（1）速度）。

PS：即使集合没有顺序，也可以保证顺序，因为循环遵循tuples而不是lst2的顺序。

Answer 2

您可以使用集合：

>>> {tuple(lst1[i:i+3]) for i in range(0,len(lst1)-2)} & {tuple(lst2[i:i+3]) for i in range(0,len(lst2)-2)}
{('Tom', 'Sam', 'Jill')}

Answer 3

您可以使用常规表达式：

import re

re.search("((?:\w+ ){3}).*\\1"," ".join(lst1)+","+" ".join(lst2)).group(1)

'Tom Sam Jill'

Answer 4

n = 3

stringlst2 = '#'.join(lst2)
for ngram in [lst1[i:i+n] for i in range(len(lst1)-n+1)]:
    if '#'.join(ngram) in stringlst2:
        print(ngram)
        break

Python：找到存在于两个列表中的给定长度的公用子列表

4 个答案: