[英]Reading csv header white space and case insensitive
是否有可能讀取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'.strip().lower()
print(lineDict[column_A])
有任何想法嗎?
您可以重新定義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'}
覆蓋DictReader.fieldnames
屬性怎么樣?
class MyDictReader(DictReader):
@property
def fieldnames(self):
return [field.strip().lower() for field in super(MyDictReader, self).fieldnames]
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.