读取csv标头空白且不区分大小写 [英] Reading csv header white space and case insensitive
本文介绍了读取csv标头空白且不区分大小写的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
是否可以读取CSV文件空白的标题并且不区分大小写?现在,我像这样使用csv.dictreader
:
Is there a possibility to read the header of a CSV file white space and case insensitive? As for now I use csv.dictreader
like this:
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文件有时在多余的空格或大写字母方面有所不同,例如
As you can see from the code, my csv files sometimes differ in extra white space or capital letters, for example
- "column_A"
- "column_A"
- "Column_A"
- "Column_A"
- ...
我想使用这样的东西:
column_A = ' Column_A'.strip().lower()
print(lineDict[column_A])
有什么想法吗?
推荐答案
如何覆盖DictReader.fieldnames
属性?
class MyDictReader(DictReader):
@property
def fieldnames(self):
return [field.strip().lower() for field in super(MyDictReader, self).fieldnames]
这篇关于读取csv标头空白且不区分大小写的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!
查看全文