2017-03-03 78 views
0

返回行數我有一個CSV文件,其內容:發現價值,價值

scenario1,5,dosomething 
scenario2,10,donothing 
scenario3,8,dosomething 
scenario4,5,donothing 

我想借一個變量的內容,首先看它是否是第一列,如果這是真的 - 我想獲取它的行號和整個行內容。 csv的第1列中不會有重複的值。

我可以部分做第一步,即查找變量是否在csv中,返回整行。

import csv 
filename = csv.reader(open('/file.csv', "rb"), delimiter=",") 
v = 'scenario1' 
for row in configfile: 
    if 'v' in row[0]: 
     print row 

我收到的結果將是:

['scenario1','5','dosomething'] 

但我需要援助與第二部分請。這是查找行號。

回答

1

試試這個:

import csv 
with open("ooo.csv", "r") as f: 
    reader = csv.reader(f) 
    for line_num, content in enumerate(reader): 
     if content[0] == "scenario1": 
      print content, line_num + 1 

或者不CSV模塊:

with open("ooo.csv") as f: 
    for l, i in enumerate(f): 
     data = i.split(",") 
     if data[0] == "scenario1": 
      print data, l + 1 

輸出:

['scenario1', '5', 'dosomething'] 1 
+0

完美 - 謝謝你,我用的第一個場景與csv模塊。 –