有没有办法通过列表理解来做到这一点？

Question

这是我写的代码：

x=['U','D','D','D','U','D','U','U']
y=[]
for i in x:
    if i=='U':
        y.append(1)
    else:
        y.append(-1)
print(y)

用 1 代替 U，用 -1 代替 D。 我们可以使用列表理解来做同样的事情吗？

Answer 1

是的，这很容易。 您将列表推导式与三元表达式相结合。

y = [1 if i=='U' else -1 for i in x]