需要使用輸入

Question

import csv 

file_reader = csv.DictReader(open('FILENAME.CSV','r')) 

for row in file_reader: 

    print(row) 

animal_input=input("What kind of animal?") 

print("To buy all those animals costs:"PRICE*QUANTITY) 

我需要能夠在input()鍵入動物的名字Excel文件內乘同一列值，因此價格x數量將會出現。但是，該數據表位於Excel文件中，因此我不知道如何從input()引用Excel文件中的「ANIMAL」列，也不知道如何包含價格和數量。

ANIMAL   PRICE QUANTITY 
ANTEATER   5   4 
BEAR    3   4 
CAT    3   4 
DOG    2   3 
ECHIDNA   2   2 

Answer 1

也許使用一個字典，並在閱讀時計算結果。

import csv 
import StringIO 

csv_pretend_file = """ANIMAL,PRICE,QUANTITY 
ANTEATER,5,4 
BEAR,3,4 
CAT,3,4 
DOG,2,3 
ECHIDNA,2,2""" 

animals = {} 

for row in csv.DictReader(StringIO(csv_pretend_file)): 
    animals[row['ANIMAL']] = int(row['PRICE']) * int(row['QUANTITY']) 


animal_input=input("What kind of animal?") 
print("To buy all those animals costs: {}".format(animals[animal_input])) 

演示：

What kind of animal?ANTEATER 


To buy all those animals costs: 20