有沒有辦法通過列表理解來做到這一點？

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]