python - 使用 python 查找 CSV 中的列号

Question

我想匹配 CSV 文件中的某个字符串，并返回 CSV 文件中该字符串的列，例如

import csv
data = ['a','b','c'],['d','e','f'],['h','i','j']

例如，我正在查找单词 e，我希望它返回第二列中的 [1]。

Answer 1

使用csv.reader对象和enumerate函数的解决方案(获取键/值序列):

def get_column(file, word):
    with open(file) as csvfile:
        reader = csv.reader(csvfile)
        for row in reader:
            for k,v in enumerate(row):
                if v == word:
                    return k  # immediate value return to avoid further loop iteration

search_word = 'e'
print(get_column("data/sample.csv", search_word))  # "data/sample.csv" is an exemplary file path

输出:

1