python 使用csv模塊讀寫csv格式文件的示例
import csvclass 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文件的資料請關注好吧啦網其它相關文章!
相關文章:
