Python標準庫之csv(1)

Python標準庫之csv(1)

1.Python處理csv文件之csv.writer()

import csv

def csv_write(path,data):
    with open(path,'w',encoding='utf-8',newline='') as f:
        writer = csv.writer(f,dialect='excel')
        for row in data:
            writer.writerow(row)
    return True

調用上面的函數

data = [
    ['Name','Height'],
    ['Keys','176cm'],
    ['HongPing','160cm'],
    ['WenChao','176cm']
]

csv_write('test.csv',data)

運行結果

 

2.Python處理csv文件之csv.reader()

def csv_read(path):
    data = []
    with open(path,'r',encoding='utf-8') as f:
        reader = csv.reader(f,dialect='excel')
        for row in reader:
            data.append(row)
    return data

調用上面的函數

data = csv_read('test.csv')
print(data)

運行結果

[['Name', 'Height'], ['Keys', '176cm'], ['HongPing', '160cm'], ['WenChao', '176cm']]

3.Python處理csv文件之csv.DictWriter()

def csv_dict_write(path,head,data):
    with open(path,'w',encoding='utf-8',newline='') as f:
        writer = csv.DictWriter(f,head)
        writer.writeheader()
        writer.writerows(data)
        return True

調用上面的函數

head = ['Name','Age']

data = [
    {'Name':'Keys', 'Age':28},
    {'Name':'HongPing', 'Age':29},
    {'Name':'WenChao', 'Age':15}
]

csv_dict_write('test2.csv',head,data)

運行結果

 

4.Python處理csv文件之csv.DictReader()

def csv_dict_read(path):
    with open(path,'r',encoding='utf-8') as f:
        reader = csv.DictReader(f,dialect='excel')
        for row in reader:
            print(row['Name'])

調用上面的函數

 

csv_dict_read('test2.csv')

 

運行結果

Keys
HongPing
WenChao