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檔案的資料請關注我們其它相關文章！