python学习——CSV模块

CSV文件

def csv_write(fileName):
    with open(fileName, 'w+', newline='') as fp:
        #以元组/列表形式写入CSV文件
        info1 = ('name', 'age')
        info2 = ('lily', 23)
        info3 = ['rose', 21]
        csvwriter = csv.writer(fp)
        csvwriter.writerows((info1, info2, info3))#writerows(数据)以list/duple格式传入，一次写入多行，每行以逗号结尾
       

        #以字典形式写入CSV文件
        info1 = {'name': 'tom', 'age': 18}
        info2 = {'name': 'lily', 'age': 23}
        info3 = {'name': 'rose', 'age': 21}
        fieldnames = ['name', 'age']#keys
        csvwriter = csv.DictWriter(fp, fieldnames=fieldnames)
        csvwriter.writeheader()#先写入表头,即keys
        csvwriter.writerows([info1, info2, info3])#writerows(list)以传入list格式，一次写入多行，每行以逗号结尾

def csv_read(fileName):
    with open(fileName, 'r') as fp:
        csv_read = csv.DictReader(fp)
        #reader = next(csv_read)

        for var in csv_read:
            my_dict = {}
            for key, value in var.items():
                my_dict[key] = value
            print(my_dict)

if __name__ == "__main__":
    fileName ='test.csv'
    # csv_read(fileName)
    # pd_read(fileName)