python - 从一个列表中查找以另一个列表中的字符串开头的字符串

Question

我有两个列表 listOne 和 listTwo

例如

listOne 可以包含“about”、“day”、“school”

listTwo 可以包含 'a'、'c'、'da'、'z'

我想找出 listOne 中以 listTwo 中元素的字符开头的所有元素。上面示例的输出是“about”和“day”

我尝试用以下代码实现它:

for elem1 in listTwo:
    for elem2 in listOne:
        if elem2.startswith(elem1):
            result.append(elem2)

但我觉得它嵌套太多了。有没有更优雅的方式用Python来实现？

Answer 1

你这样做的方式很好。易于阅读/理解等。

但是，如果您确实想要，您可以使用 itertools.product 将其压缩:

from itertools import product
result = [elem2 for elem1, elem2 in product(listTwo, listOne) if elem2.startswith(elem1)]