使用Python raw_input从CSV文件获取特定值？

Question

目前，我已经设置了这样一个CSV文件：

Element,Weight 
"Hydrogen","1" 
"Oxygen","16" 

它最终将所有部件和它们的原子量的旁边，而是将在所有是毫无意义，除非我得到这个问题首先解决。

主要Python程序如下：

import csv 
reader = csv.reader(open("chem.csv", "rb")) 

first = raw_input("Enter first element: ") 
second = raw_input("Enter second element: ") 

if first == "Hydrogen" or "hydrogen": 
    for Weight in reader: 
     print Weight 

else: 
    print "No." 

现在，你也许可以从讲，在这里我的目标是让程序显示氢的重量从CSV文件中取出， 目前。但是，目前它只是最终显示如下：

Enter first element: hydrogen 
Enter second element: oxygen 
['Element', 'Weight'] 
['Hydrogen', '1'] 
['Oxygen', '16'] 

所以，基本上，我怎样才能让这个不言而喻的氢气行，然后取从第二列中的权重值？我应该可以从那里去完成剩下的程序，但这部分让我陷入困境。

作为次要的东西，是否有可能拥有它，以便我不必在Python主程序和CSV文件中拥有本质上的元素列表？它会让杂乱减少很多，但我无法弄清楚。

Answer 1

不是迭代数据，而是以易于访问的方式来存储数据。一个字典映射名称到重量。然后，您可以只查找输入的字符串并显示结果（如果没有，则不显示）。

import csv 

#Read in data anbd store it in dictionary. 
#The element name will be stored in lowercase. 
elemen_data = { element.lower():int(weight) for element,weight in csv.reader(open("in", "rb"))} 

first = raw_input("Enter first element: ") 
second = raw_input("Enter second element: ") 

#Look for the weight of the entered element (after converting it to all lowercase) 
weight_first = elemen_data.get(first.lower()) 
weight_second = elemen_data.get(second.lower()) 

if weight_first is not None: 
    print 'First Element:', first, 'Weight:', weight_first 
else: 
    print 'First Element', first, 'not found' 

if weight_second is not None: 
    print 'Second Element:', second, 'Weight:', weight_first 
else: 
    print 'Second Element', second, 'not found' 

从你的程序逻辑分开存储你的数据是有意义的，如果你希望能够方便地更改数据而不触及程序。所以读取一个csv文件是绝对没问题的。 （只要数据具有合理的大小）

Answer 2

问题是，当您遍历csv.reader对象时，您会收到csv文件行作为行单元格值列表。所以你可以尝试下你的情况：

if first in ("Hydrogen", "hydrogen"): 
    for Weight in reader: 
     if Weight[0] == first: 
      print Weight[1] 

这不是最好的代码，但它会返回你想要的值。

Answer 3

第一个问题：

if first == "Hydrogen" or "hydrogen": 

这将不会返回你想要的。要么你想：

if first == 'Hydrogen' or first == 'hydrogen': 

或

if first in ['Hydrogen', 'hydrogen']: 

或者更好的是：

if first.lower() == 'hydrogen' 

这将适用于所有已资本化，氢气等。

然后，你可以做你想做的：

for row in reader: 
    if row[0].lower() == first.lower(): 
     print row[1] 

当然，这可以通过使用值的字典被清理了一下;你可能不想为每个元素使用if声明，但是你去了。