python - Python - 見出しを使用したファイル操作

Question

Python で .txt ドキュメントを実行し、特定の見出しを見つけて、各行の情報を印刷用のリストに入れるにはどうすればよいですか? 終了したら、別の見出しを探して、そこにある情報で同じことを行います...

Answer 1

次のような csv ファイルがあるとします。

h1,h2,h3
a,b,c
d,e,f
g,h,i

次に、あなたが要求したとおりに次のことを行います（私があなたを正しく理解していれば）

def getColumn(title,file):
    result = []
    with open(file) as f:
        headers = f.readline().split(',')
        index = headers.index(title)
        for l in f.readlines():
            result.append(l.rstrip().split(',')[index])
    return result

例えば：

print(getColumn("h1",'cf.csv') )
>>> ['a', 'd', 'g']