Python dictreader - CSV sütun adları nasıl küçük harf yapılır?

Question

Büyük harflerle sütun adlarına sahip bir CSV dosyasına sahibim. Ben csv.dictreader kullanarak veri okuyorum ama küçük harflerle sütun isimlerine ihtiyacım var.

buldum burada bu kod Accessing csv header white space and case insensitive

import csv 

class DictReaderInsensitive(csv.DictReader): 
    # This class overrides the csv.fieldnames property. 
    # All fieldnames are without white space and in lower case 

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

    def __next__(self): 
     # get the result from the original __next__, but store it in DictInsensitive 

     dInsensitive = DictInsensitive() 
     dOriginal = super(DictReaderInsensitive, self).__next__() 

     # store all pairs from the old dict in the new, custom one 
     for key, value in dOriginal.items(): 
      dInsensitive[key] = value 

     return dInsensitive 

class DictInsensitive(dict): 
    # This class overrides the __getitem__ method to automatically strip() and lower() the input key 

    def __getitem__(self, key): 
     return dict.__getitem__(self, key.strip().lower()) 

Sorunum

datafile = open(self.ifs_data_file,'rU') 
     csvDict = DictReaderInsensitive(datafile) 
     for row in csvDict: 
      print row 
      #self.db.ifs_data.insert(**row) 
      #self.db.commit() 

ile bu çalıştırdığınızda bu hatayı

Traceback (most recent call last): 
    File "D:\Development\python\supplier_review\supplier_review.py", line 239, in update_ifs_data 
    for row in csvDict: 
    File "D:\Python27_5\lib\csv.py", line 103, in next 
    self.fieldnames 
    File "D:\Development\python\supplier_review\supplier_review.py", line 288, in fieldnames 
    return [field.strip().lower() for field in super(DictReaderInsensitive, self).fieldnames] 
TypeError: must be type, not classobj 

Answer 1

DictReader olsun ki eski tarzıdır Nesne, yani super() hiç çalışmıyor. Ebeveyn sınıfındaki property nesnesine doğrudan erişmeniz gerekir.

class DictReaderInsensitive(csv.DictReader): 
    # This class overrides the csv.fieldnames property. 
    # All fieldnames are without white space and in lower case 

    @property 
    def fieldnames(self): 
     return [field.strip().lower() for field in csv.DictReader.fieldnames.fget(self)] 

    def next(self): 
     return DictInsensitive(csv.DictReader.next(self)) 

Demo:

>>> example = '''\ 
... foo,Bar,BAZ 
... 42,3.14159,Hello world!'''.splitlines() 
>>> csvDict = DictReaderInsensitive(example) 
>>> row = next(csvDict) 
>>> print row 
{'bar': '3.14159', 'foo': '42', 'baz': 'Hello world!'} 
>>> row['BAZ'] 
'Hello world!' 

Answer 2

Sen DictReader geçirmeden önce dosyanın ilk satırını küçük harfe olabilir:

import csv 
import itertools 

def lower_first(iterator): 
    return itertools.chain([next(iterator).lower()], iterator) 

with open(ifs_data_file, 'rU') as datafile: 
    csvDict = csv.DictReader(lower_first(datafile)) 
    for row in csvDict: 
     print row  

Python 2'de, sen .next() yöntemi değil .__next__() geçersiz kılmak istiyorsanız

Answer 3

Daha basit bir yaklaşım için, aşağıdaki gibi, sözlüğünüze erişmeden önce DictReader.fieldnames özniteliğini basitçe güncelleyebilirsiniz:

>>> f = open('example-x-y-time.csv', 'rb') 
>>> reader = csv.DictReader(f) 
>>> reader.fieldnames 
['Latitude', 'Longitude', 'Date'] 
>>> print next(reader) 
{'Latitude': '44.8982391', 'Date': '2004-07-12', 'Longitude': '-117.7791061'} 
>>> reader.fieldnames = [name.lower() for name in reader.fieldnames] 
>>> print next(reader) 
{'latitude': '44.6637001', 'date': '1964-04-03', 'longitude': '-123.5997009'}