第十五节 CSV文件读写

import csv


def csv_rd():
    with open(r'E:anacondatestPythonDataaaaaa.csv','r',encoding='utf-8') as fp:

        readers = csv.reader(fp)
        print(type(readers))
        # next(readers)
        # readers是一个迭代器,遍历这个迭代器返回的是一个列表
        for x in readers:
            name = x[2]
            score = x[3]
            print(name,score)

def csv_rd2():
    with open(r'E:anacondatestPythonDataaaaaa.csv', 'r', encoding='utf-8') as fp:
        readers = csv.DictReader(fp)
        #readers不会包含标题行的数据
        # readers是一个迭代器,遍历这个迭代器返回的是一个字典
        for x in readers:
            print(x['学校名称'],x['总分'])

headers = ['name','age','height']
values = [
    ('李梅',20,167),
    ('lili',19,173),
    ('tom',18,185)
]

def write_csv1():
    with open(r'demo1.csv', 'w',encoding='utf-8',newline='') as fp:
        #newline=''参数，默认写入一行数据以后换行
        writers = csv.writer(fp)
        #写入头部
        writers.writerow(headers)
        #写入数据
        writers.writerows(values)

def write_csv2():
    value = [
        {'name':'张三','age':18,'height':180},
        {'name': '王五', 'age': 19, 'height': 175},
        {'name': '李四', 'age': 20, 'height': 168}
    ]
    with open(r'demo2.csv', 'w',encoding='utf-8',newline='') as fp:
        #newline=''参数，默认写入一行数据以后换行
        writers = csv.DictWriter(fp,headers)
        #调用writeheader()方法写入变量名
        writers.writeheader()
        writers.writerows(value)
        
if __name__ == '__main__':
    write_csv2()