python读写csv文件

普通方法读取：

 def get_data1(self, path,header):
        """
        :param path: csv文件路径
        :param header: 表头
        :return: 由多组数据组成的列表
        """
        list = []
        with open(path) as file:
            data = file.readlines()
            for line in data:
                line = line.strip().split(',')  # csv文件是以，作为分隔符的文本文件
                list.append(line)
            if header:
                list = list[1:]
        return list

if __name__ == '__main__':
    mail_test = MailTest()
    user_list = mail_test.get_data1(r'C:\DATA\selenium\mail_login.csv',1)
    print(user_list)

用CSV标准库读取：

import csv


def get_data2(self,path,header):
        """
        :param path: csv文件路径
        :param header: 表头
        :return: 由多组数据组成的列表
        """
        list = []
        csv_reader = csv.reader(open(path))
        for row in csv_reader:
            list.append(row)
        if header:
            list = list[1:]
        return list

if __name__ == '__main__':
    mail_test = MailTest()
    user_list = mail_test.get_data2(r'C:\DATA\selenium\mail_login.csv',1)
    print(user_list)

用pandas读取：

import pandas as pd


def get_data2(self,path,header):
    read_pd = pd.read_csv(path)
    data = read_pd.values
    return data


if __name__ == '__main__':
    mail_test = MailTest()
    user_list = mail_test.get_data2(r'C:\DATA\selenium\mail_login.csv',1)
    print(user_list)

用CSV标准库写入：

import csv
data = [['zyd','23'],['zxf','22']]
f = open(r'C:\DATA\selenium\mail_login.csv','a',newline='')
csv_write = csv.writer(f)
for row in data:
    csv_write.writerow(row)
f.close()