将循环块转换为列表理解

Question

我想转换这个片段

for val in G[k]:
    if val in span[k]:
        result.append((val,col))
    elif val in G[k] and val not in span[k] and S[val][k] != 'col':
        result.append((val,row))

进入列表理解。 但这给了我错误：

[(val,col) if val in span[k] else (val,row) if val in G[k] and val not in span[k] and S[val][k] != 'col' for val in G[k]]

那么正确的语法是什么，如果有的话

Answer 1

更改

[(val,col) if val in span[k] else (val,row)
 if val in G[k] and val not in span[k] and S[val][k] != 'col'
 for val in G[k]]

至

[(val,col) if val in span[k] else (val,row)
 for val in G[k]
 if val in G[k] and val not in span[k] and S[val][k] != 'col' ]

经验法则如果有if之前for在理解，它必须有一个else 。

Answer 2

您正在使用没有else部分的条件表达式; 这是一个语法错误。

正确的翻译是：

[(val,col) if val in span[k] else (val,row) for val in G[k] if val in span[k] or S[val][k] != 'col']

例如，过滤掉任何不首先匹配你的两个条件，并且不产生添加到输出的东西，值的两个分支之间进行选择。

我简化了条件; 有一个在第二表达一些多余的测试（ val in G[k]是总是如此为一个循环在G[k] ，并且如果val in span[k] 不是 True ，则逆val not in span[k]是肯定True的好，并不需要再次进行测试。

就个人而言，我发现显式for循环更具可读性，但你至少可以用同样的方式简化它：

for val in G[k]:
    if val in span[k]:
        result.append((val,col))
    elif S[val][k] != 'col':
        result.append((val,row))