5

CSVファイルの空白と大文字と小文字を区別しないヘッダーを読み取る可能性はありますか？今のところ私はcsv.dictreaderこのように使用します：

import csv
csvDict = csv.DictReader(open('csv-file.csv', 'rU'))

# determine column_A name
if 'column_A' in csvDict.fieldnames:
    column_A = 'column_A'
elif ' column_A' in csvDict.fieldnames:
    # extra space
    column_A = ' column_A'
elif 'Column_A' in csvDict.fieldnames:
    # capital A
    column_A = 'Column_A'

# get column_A data
for lineDict in csvDict:
    print(lineDict[column_A])

コードからわかるように、たとえば、私のcsvファイルは余分な空白や大文字が異なる場合があります

  • "column_A"
  • "column_A"
  • 「Column_A」
  • 「Column_A」
  • ..。

私はこのようなものを使いたい：

    column_A = ' Column_A'.strip().lower()
    print(lineDict[column_A])

何か案は？

4

2 に答える 2

13

あなたは再定義することができますreader.fieldnames：

import csv
import io

content = '''column_A " column_B"
1 2'''
reader = csv.DictReader(io.BytesIO(content), delimiter = ' ')
reader.fieldnames = [field.strip().lower() for field in reader.fieldnames]
for line in reader:
    print(line)

収量

{'column_b': '2', 'column_a': '1'}
于 2012-10-17T12:16:15.400 に答える
7

オーバーライドDictReader.fieldnamesプロパティはどうですか？

class MyDictReader(DictReader):

    @property
    def fieldnames(self):
        return [field.strip().lower() for field in super(MyDictReader, self).fieldnames]
于 2012-10-17T12:14:54.943 に答える