CSV(Comma-Separated Values)即逗號(hào)分隔值，一種以逗號(hào)分隔按行存儲(chǔ)的文本文件，所有的值都表現(xiàn)為字符串類(lèi)型（注意：數(shù)字為字符串類(lèi)型）。

如果CSV中有中文，應(yīng)以u(píng)tf-8編碼讀寫(xiě)，如果要支持Excel查看，應(yīng)是要用utf-8 with bom格式及utf-8-sig

Python3操作CSV文件使用自帶的csv包

reader=csv.reader(f, delimiter=',')：用來(lái)讀取數(shù)據(jù)，reader為生成器，每次讀取一行，每行數(shù)據(jù)為列表格式，可以通過(guò)delimiter參數(shù)指定分隔符

writer=csv.writer(f)：用來(lái)寫(xiě)入數(shù)據(jù)，按行寫(xiě)入，writer支持writerow(列表)單行寫(xiě)入，和writerows(嵌套列表)批量寫(xiě)入多行，無(wú)須手動(dòng)保存。

當(dāng)文件中有標(biāo)題行時(shí)，可以使用header=next(reader)先獲取到第一行的數(shù)據(jù)，再進(jìn)行遍歷所有的數(shù)據(jù)行。
寫(xiě)入時(shí)，可以先使用writer.writerow(標(biāo)題行列表)，寫(xiě)入標(biāo)題行，再使用writer.writerows(多行數(shù)據(jù)嵌套列表)，寫(xiě)入多行數(shù)據(jù)（也可以逐行寫(xiě)入）。

數(shù)據(jù)文件data.csv:

name,password
abc,123456
張五,123#456
張#abc123,123456
666,123456
a b,123456

讀取示例：

import csv
with open('data.csv', encoding='utf-8') as f:
  reader = csv.reader(f)
  header = next(reader)
  print(header)
  for row in reader:
    print(row)

注意：reader必須在文件打開(kāi)的上下文中使用，否則文件被關(guān)閉后reader無(wú)法使用
所有的數(shù)字被作為字符串，如果要使用數(shù)字格式，應(yīng)使用int()/float()做相應(yīng)轉(zhuǎn)換

寫(xiě)入示例：

import csv
header = ['name', 'password', 'status']
data = [
  ['abc', '123456', 'PASS'],
  ['張五', '123#456', 'PASS'],
  ['張#abc123', '123456', 'PASS'],
  ['666', '123456', 'PASS'],
  ['a b', '123456', 'PASS']
]
with open('result.csv', 'w', encoding='utf-8', newline='') as f:
  writer = csv.writer(f)
  writer.writerow(header)
  writer.writerows(data)

注意，打開(kāi)文件時(shí)應(yīng)指定格式為w, 文本寫(xiě)入，不支持wb,二進(jìn)制寫(xiě)入，當(dāng)然，也可以使用a/w+/r+

打開(kāi)文件時(shí)，指定不自動(dòng)添加新行newline='',否則每寫(xiě)入一行就或多一個(gè)空行。

如果想寫(xiě)入的文件Excel打開(kāi)沒(méi)有亂碼，utf-8可以改為utf-8-sig。

結(jié)果展示，result.csv內(nèi)容：：

name,password,status
abc,123456,PASS
張五,123#456,PASS
張#abc123,123456,PASS
666,123456,PASS
a b,123456,PASS

使用字典格式的數(shù)據(jù)：DictReader, DictWriter#

注意數(shù)據(jù)必須有標(biāo)題行時(shí)才能使用

reader=csv.DictReader(f)：直接將標(biāo)題和每一列數(shù)據(jù)組裝成有序字典（OrderedDict）格式，無(wú)須再單獨(dú)讀取標(biāo)題行
writer=csv.DictWriter(f, 標(biāo)題行列表)：寫(xiě)入時(shí)可使用writer.writeheader()寫(xiě)入標(biāo)題，然后使用writer.writerow(字典格式數(shù)據(jù)行)或write.writerows(多行數(shù)據(jù))

讀取示例：

import csv
with open('data.csv', encoding='utf-8') as f:
  reader = csv.DictReader(f)
  for row in reader:
    print(row['name'], row['password'])

寫(xiě)入示例：

import csv

header = ['name', 'password', 'status']

data = [
  {'name':'abc', 'password':'123456', 'status':'PASS'},
  {'name':'張五', 'password':'123#456', 'status':'PASS'},
  {'name':'張#abc123', 'password':'123456', 'status':'PASS'},
  {'name':'666', 'password':'123456', 'status':'PASS'},
  {'name':'a b', 'password':'123456', 'status':'PASS'}
]


with open('result2.csv', 'w', encoding='utf-8', newline='') as f:
  writer = csv.DictWriter(f, header)
  writer.writeheader()
  writer.writerows(data)

寫(xiě)入結(jié)果，同result.csv

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

最新評(píng)論