importcsvclass
handlecsv:
'''csv檔案處理類
'''def
__init__
(self, filename):
'''構造器
:param filename: csv檔名
'''self.filename =filename
defget_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:
return
list_dict
defget_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("
資料型別錯誤,請確認!
")
python3使用csv模組讀寫csv檔案
讀取csv檔案 import csv 開啟檔案,用with開啟可以不用去特意關閉file了,python3不支援file 開啟檔案,只能用open with open csv r encoding utf 8 as csvfile 讀取csv檔案,返回的是迭代型別 read csv.reader c...
Python 使用 csv 模組讀寫 csv 檔案
目錄 一 讀取 csv 檔案 二 寫入 csv 檔案 coding utf 8 import csv import sys def read csv file filename with open filename,r as f csv reader csv.reader f header row ...
python使用csv讀寫CSV檔案
檔案的讀 import csv with open test.csv r as csv file reader csv.reader csv file for line in reader print line 檔案的寫 import csv 表頭 file header houseinfo hou...