python - 在 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 = '一二三四'.split()

indexNum = a.index('三')

打印 indexNum # 为 2