Python学习笔记--读写csv文件

"""读写csv文件"""
# 现有五个学生三门课程的考试成绩需要保存到一个CSV文件中，要达成这个目标，
# 可以使用Python标准库中的csv模块，该模块的writer函数会返回一个csvwriter对象，
# 通过该对象的writerow或writerows方法就可以将数据写入到CSV文件中
import csv
import random


# pandas中封装了名为read_csv和to_csv的函数用来读写CSV文件，
# 其中read_CSV会将读取到的数据变成一个DataFrame对象，而这个对象就是pandas库中最重要的类，
# 它封装了一系列的方法用于对数据进行处理（清洗、转换、聚合等）；
# 而to_csv会将DataFrame对象中的数据写入CSV文件，完成数据的持久化。

# 写文件
def write_to_csv():
    with open('score.csv', 'w', encoding='gbk') as file:
        writer = csv.writer(file)
        writer.writerow(['姓名', '语文', '数学', '英语'])
        names = ['关羽', '张飞', '赵云', '马超', '黄忠']
        for i in range(5):
            verbal = random.randint(50, 100)
            math = random.randint(40, 100)
            english = random.randint(30, 100)
            writer.writerow([names[i], verbal, math, english])


# write_to_csv()

# 读文件
def read_from_csv():
    with open('score.csv', 'r') as file:
        reader = csv.reader(file, delimiter='|')
        for line in reader:
            # print(reader.line_num, end='\t')
            for elem in line:
                print(elem, end='\t')
            print()

# read_from_csv()