高级 Python 列表理解

Question

给定两个列表：

chars = ['ab', 'bc', 'ca']
words = ['abc', 'bca', 'dac', 'dbc', 'cba']

如何使用列表推导式通过以下条件生成过滤的 words 列表：假定每个单词的长度为 n 和 chars< /code> 的长度也是 n，过滤后的列表应仅包含每个 i 个字符位于第 i 个字符中的单词words 中的字符串。

在这种情况下，我们应该得到 ['abc', 'bca'] 结果。

（如果这对任何人来说都很熟悉，这是之前 Google Code Jam 中的问题之一）

Answer 1

>>> [word for word in words if all(l in chars[i] for i, l in enumerate(word))]
['abc', 'bca']

Answer 2

[w for w in words if all([w[i] in chars[i] for i in range(len(w))])]

Answer 3

使用 zip:

[w for w in words if all([a in c for a, c in zip(w, chars)])]

或使用 enumerate:

[w for w in words if not [w for i, c in enumerate(chars) if w[i] not in c]]

Answer 4

使用 index 可以正常工作：

[words[chars.index(char)] for char in chars if char in words[chars.index(char)]]

我错过了什么吗？

Answer 5

更简单的方法：

yourlist = [ w for w in words for ch in chars if w.startswith(ch) ]

Answer 6

为什么这么复杂？这也有效：

[words[x] for x in range(len(chars)) if chars[x] in words[x]]