讀取csv頭部空白區域和不區分大小寫

Question

是否有可能讀取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]) 

任何想法？

Answer 1

如何覆蓋DictReader.fieldnames屬性？

class MyDictReader(DictReader): 

    @property 
    def fieldnames(self): 
     return [field.strip().lower() for field in super(MyDictReader, self).fieldnames] 

Answer 2

你可以重新定義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'}