python 使用csv模块读写csv格式文件的示例
import csv class HandleCsv: ''' csv文件处理类 ''' def __init__(self, filename): ''' 构造器 :param filename: csv文件名 ''' self.filename = filename def get_data(self): ''' 获取csv中所有数据 :return: 嵌套字典的列表 ''' with open(self.filename, mode='r', encoding='utf-8') as f: cb = csv.reader(f) # 实例化reader对象 header = next(cb) # 获取表头,并将指针转向下一行 list_dict = [] for row in cb: list_dict.append(dict(zip(header, row))) return list_dict def get_one_row(self, row): ''' 获取单行数据 :param row: 指定的行号 :return: 对应行号的数据 ''' return self.get_data()[row - 1] def write_csv(self, headers, values, data_type, mode='w'): ''' 写入数据到csv到文件中 :param headers: 表头:列表类型 :param values: 表数据:1.嵌套元组的列表;2.嵌套字典的列表 :param data_type: 传入的数据类型:1.'tuple';2.'dict' :param mode: 写入方式,默认写入“w” :return: ''' with open(file=self.filename, mode=mode, encoding='utf-8', newline='') as f: if data_type == 'tuple': writer = csv.writer(f) # 实例化writer对象 writer.writerow(headers) # 写入表头 writer.writerows(values) # 写入数据 elif data_type == 'dict': writer = csv.DictWriter(f, headers) # 实例化DictWriter对象 writer.writeheader() # 写入表头 writer.writerows(values) # 写入数据 else: print("数据类型错误,请确认!")
以上就是python 使用csv模块读写csv格式文件的示例的详细内容,更多关于python 读写csv文件的资料请关注我们其它相关文章!
赞 (0)