在csv中搜索字符串并保存该列

Question

使用pythons csv模块，我正在尝试打开CSV文件，搜索特定字符串

在该文件中，一旦找到，就存储该列。这是我现在阅读专栏的方式，但我

并不总是知道我想要的特定列号，所以我需要搜索一列

改为命名。

with open('Work.csv','r') as f:
 reader = csv.reader(f)
 reader.next()
  for row in reader:
    for (i ,v) in enumerate(row):
      columns[i].append(v)

我的csv看起来像：

Default  Names  
  0        1
  2        3

Answer 1

import csv
columns = []                        #save the columns in this list
with open('myfile.csv','r') as f:
 reader = csv.reader(f, delimiter='\t')
 ind = next(reader).index('Default')   #find the index of 'Default' in the header
 for row in reader:
     columns.append(row[ind])

Answer 2

从行创建列表并使用列表中的索引函数。

例如

a ='one two three four'.split（）

indexNum = a.index（'three'）

print indexNum＃，即2