python處理csv文件

Python讀取csv文件

1、使用列表的方式讀取

import csv

with open('1.csv', 'r', encoding='utf-8') as fp:
    # reader是一個迭代器
    reader = csv.reader(fp)
    next(reader)
    # next(reader)
    for x in reader:
        print(x)

reader是一個迭代器，因此使用next方法，可以去掉表頭

2、使用字典的方式讀取

with open('1.csv', 'r') as fp:
    # 使用DictReader創建的對象，不會包行標題那行的數據
    reader = csv.DictReader(fp)
    #reader是一個迭代器，遍歷這個迭代器，返回的是一個字典
    for x in reader:
        print(x)
        print(x.get('姓名'))
結果
OrderedDict([('姓名', '小劉'), ('年齡', '25'), ('性別', '男')])
小劉
OrderedDict([('姓名', '小羅'), ('年齡', '26'), ('性別', '女')])
小羅

Python寫入csv文件

1、使用列表的方式寫入

header = ['姓名', '年齡', '性別']
contents = [('小劉', '25', '男'), ('小羅', '26', '女')]
with open('2.csv', 'w', newline='') as fp:
    writer = csv.writer(fp)
    writer.writerow(header)
    writer.writerows(contents)

newline=’’
2、使用字典的方式寫入

header = ['姓名', '年齡', '性別']
contents = [('小劉', '25', '男'), ('小羅', '26', '女')]
with open('2.csv', 'w', newline='', encoding='utf-8') as fp:
    writer = csv.writer(fp)
    writer.writerow(header)
    writer.writerow(header)
    writer.writerows(contents)

header = ['姓名', '年齡', '性別']
values = [
    {'姓名': '小劉', '年齡': '25', '性別': '男'},
    {'姓名': '小羅', '年齡': '26', '性別': '女'},
]
with open('3.csv', 'w', encoding='utf-8', newline='') as fp:
    writer = csv.DictWriter(fp, header)
    # 寫入表頭數據的時候，需要調用writeheader方法
    writer.writeheader()
    writer.writerows(values)