三元运算符python3.5的列表推导

Question

我有一个名为givenProductions 的字符串列表。每个字符串都有一个大写字母和“-”，并且可能包含一个小写字母。

示例：givenProductions 可以是['S-AA', 'A-a', 'A-b']

现在我想填充两组：

1. 终端（仅包含来自givenProductions的小写字母）和
2. 非终端（仅包含来自givenProductions 的大写字母）

只有一行

我试过这个...

terminals.append(ch) if (ch >= 'a' and ch <= 'z') else nonTerminals.append(ch) if (ch != '-') else print() for ch in prod for prod in givenProductions

导致语法错误

File "<stdin>", line 2
    terminals.append(ch) if (ch >= 'a' and ch <= 'z') else nonTerminals.append(ch) if (ch != '-')   else print ('-') for ch in prod  for prod in givenProductions
                                                                                                                       ^
SyntaxError: invalid syntax

正确的写法是什么？

Answer 1

如果您不关心结果，那么列出推导式绝对没有用。把它写成一个普通的for循环：

for prod in givenProductions:
    for ch in prod:
        if ch >= 'a' and ch <= 'z':
            terminals.append(ch)
        elif ch != '-':
            nonTerminals.append(ch)

请注意，两个for 循环的顺序发生了变化！如果您真的想要使用列表推导式，您也必须这样做。不需要打印，只需用None 完成三元（print() 无论如何都会产生）。此外，列表理解需要方括号 ([])：

>>> givenProductions = ['S-AA', 'A-a', 'A-b']
>>> terminals, nonTerminals = [], []
>>> [terminals.append(ch) if (ch >= 'a' and ch <= 'z') else nonTerminals.append(ch) if (ch != '-') else None for prod in givenProductions for ch in prod]
>>> terminals, nonTerminals
>>> print(terminals, nonTerminals)
['a', 'b'] ['S', 'A', 'A', 'A', 'A']

请注意，这会创建并丢弃None 元素的列表。使用集合解析 ({}) 的内存效率更高，但 CPU 效率更低。

>>> waste = {terminals.append(ch) if (ch >= 'a' and ch <= 'z') else nonTerminals.append(ch) if (ch != '-') else None for prod in givenProductions for ch in prod}
>>> print(waste)
{None}