如何使用列表理解使这个循环 Python 代码更有效？

Question

我无法将下面显示的代码逻辑转换为使用 Python 理解。 谁能帮我吗？

前提

lst1是匹配词对的列表， lst2是匹配词六重奏的列表。 我试图将lst2中的任意两个条目“缝合”在一起，以使用lst1中的单词对形成十二个单词的循环，方法是将lst2条目中的开头和结尾单词与条目lst1 。

输入

lst1和lst2是 2D 列表，格式如下（带有一些示例数据）：

lst1 = [['RUNS', 'SHORT'],
        ['HIGH', 'HORSE'],
        ['TRUE', 'FALSE'],
        ['KEEP', 'HOUSE']
       ]

lst2 = [['SHORT', 'FILM', 'PROP', 'PLANE', 'RIDE', 'HIGH'],
        ['FRONT', 'DOOR', 'NAIL', 'SALON', 'DECK', 'HAND'],
        ['HORSE', 'BACK', 'FLIP', 'PHONE', 'HOME', 'RUNS']
       ]

我的代码

我想出的绝对非 Pythonic 代码是：

ans = []

for i in lst1:
    for j in lst2:
        for k in lst2:
            if j[-1] == i[0] and k[0] == i[1]:
                item = j + k
                for m in lst1:
                    if m[1] == j[0] and m[0] == k[-1]:
                        ans.append(item)
print(ans)

# becasue by definition j[0] = item[0] and k[-1] = item[-1],
# the last if statement can also be written as:

#   if m[1] == item[0] and m[0] == item[-1]:

Output

此代码在上述数据上运行产生以下 output：

[['HORSE', 'BACK', 'FLIP', 'PHONE', 'HOME', 'RUNS',
  'SHORT', 'FILM', 'PROP', 'PLANE', 'RIDE', 'HIGH'],
 ['SHORT', 'FILM', 'PROP', 'PLANE', 'RIDE', 'HIGH',
  'HORSE', 'BACK', 'FLIP', 'PHONE', 'HOME', 'RUNS']
]

请注意，对于 ans 中的每个条目， [ ans [ans[i][-1], ans[i][0]]都在lst2中，因为需要再次“循环”到开头以形成一个单词圈。 由于单词循环的循环特性， ans中的两个条目是同一个单词循环，并且本质上是重复的条目。 我需要返回 go 并使用我还没有弄清楚的其他代码清理这个重复。

（旁注：这些嵌套for循环在具有数万个条目的列表上运行非常慢。关于如何使用不同的逻辑加快速度的任何想法？单词列表可能有重复的条目需要保留，因此dict()或set()不起作用。）

先感谢您。 我和 Python 有点菜鸟，可以使用我能得到的所有帮助。

炼金术士

Answer 1

这就是我一直在寻找的：

ans = [ j + k  for i in corner
               for w in corner
               for j in chain
               for k in chain
               if j[-1] == i[0] and k[0] == i[1]
               if w[1] == j[0] and w[0] == k[-1]
               ]
print(ans)

它稍微快一点。 感谢您的所有评论。

炼金术士