python中操作csv文件

一、简介
在写python脚本时，常常我们需要对excel文件进行处理，都知道，excel文件可以转为
csv文件，cvs文件相对简单轻量。这里将介绍使用python对csv文件进行操作的方法。
二、代码
这里仍然直接上代码，解释请查看注释。

#! /usr/bin/env python
# -*- coding:utf8 -*-
import csv

# 以位置索引的方式读csv
def read_csv(csv_file):
    with open(csv_file) as f:
        reader = csv.reader(f)
        for row in reader:
            # print(row)
            print(row[0])


# 以位置索引的方式写csv
def write_csv(csv_file):
    with open(csv_file, mode='w') as f:
        writer = csv.writer(f)
        writer.writerow(['yes'] * 5 + ['bug'])
        writer.writerow(['no'] * 5)
        writer.writerow(['no', 'go', 'a,d', 'b'])


# 以map的方式读csv
def read_dict_csv(csv_file):
    with open(csv_file) as f:
        # 定义key
        field_names = ['id', 'name']
        # 写入数据
        reader = csv.DictReader(f, fieldnames=field_names)
        for row in reader:
            # print(row)
            print(row['name'])


# 以map的方式写csv
def write_dict_csv(csv_file):
    with open(csv_file, mode='w') as f:
        # 定义key
        field_names = ['id', 'name']
        writer = csv.DictWriter(f, fieldnames=field_names)
        # 写入数据
        writer.writerow({'id': '1', 'name': 'go1'})
        writer.writerow({'id': '2', 'name': 'go2'})
        writer.writerow({'name': 'go3', 'id': '3'})


def main():
    # 以位置索引的方式读写csv
    csv_file = 't.csv'
    write_csv(csv_file)
    read_csv(csv_file)

    # 以map的方式读写csv
    csv_dict_file = 't_dict.csv'
    write_dict_csv(csv_dict_file)
    read_dict_csv(csv_dict_file)


if __name__ == '__main__':
    main()