我有一個類似下面的函數。我想要做的是讓我說我有checklist('sky high')
,那麼輸出的產品將是30,19
而不是0,30,0,19
就像我現在的代碼所做的那樣。如果我有checklist('blue sky')
,則產量將爲0
。那麼我該如何改變我的代碼呢?如果語句混淆
def checklist(argument):
data = [["sky",'79'],
["sky high",'30'],
["sky sky",'50'],
["sky high",'19']]
for row in data:
if argument == row[0]:
print row[1]
else:
print 0