python中csv格式文件如何實現寫入與讀取

這篇文章主要為大家展示了“python中csv格式文件如何實現寫入與讀取”，內容簡而易懂，條理清晰，希望能夠幫助大家解決疑惑，下面讓小編帶領大家一起研究并學習一下“python中csv格式文件如何實現寫入與讀取”這篇文章吧。

csv的簡單介紹

CSV (Comma Separated Values)，即逗號分隔值（也稱字符分隔值，因為分隔符可以不是逗號），是一種常用的文本格式，用以存儲表格數據，包括數字或者字符。很多程序在處理數據時都會碰到csv這種格式的文件。python自帶了csv模塊，專門用于處理csv文件的讀取

csv的寫入

1通過創建writer對象，主要用到2個方法。一個是writerow，寫入一行。另一個是writerows寫入多行

2使用DictWriter 可以使用字典的方式把數據寫入進去

第一種寫入方法(通過創建writer對象)

?先來說一下第一種寫入的方法：通過創建writer對象寫入（每次寫入一行）

步驟：1.創建數據和表頭2.創建writer對象3.寫表頭4.遍歷列表，將每一行數據寫入csv

代碼如下：

import csv

person = [('xxx', 18, 193), ('yyy', 18, 182), ('zzz', 19, 185)]
# 表頭
header = ['name', 'age', 'height']

with open('person.csv', 'w', encoding='utf-8') as file_obj:
    # 1:創建writer對象
    writer = csv.writer(file_obj)
    # 2:寫表頭
    writer.writerow(header)
    # 3:遍歷列表，將每一行的數據寫入csv
    for p in person:
        writer.writerow(p)

寫入完就會在當前目錄下出現一個person.csv文件，鼠標右鍵點擊show in Explorer打開person.csv查看

打開以后會發現寫入的數據中間會換行

居然：那么應該怎么解決這個問題呢

hacker：很簡單啊

只需要在寫入數據的時候加上一個參數 newline=’'為了防止換行寫入

改正后的代碼如下：

import csv

# 數據
person = [('xxx', 18, 193), ('yyy', 18, 182), ('zzz', 19, 185)]
# 表頭
header = ['name', 'age', 'height']

with open('person.csv', 'w', encoding='utf-8', newline='') as file_obj:
    # 創建對象
    writer = csv.writer(file_obj)
    # 寫表頭
    writer.writerow(header)
    # 遍歷，將每一行的數據寫入csv
    for p in person:
        writer.writerow(p)

?通過創建writer對象（一次性寫入多行）

步驟：1.創建數據和表頭2.創建writer對象3.寫表頭4.在writerows里傳入你要處理的數據

import csv

# 數據
person = [('xxx', 18, 193), ('yyy', 18, 182), ('zzz', 19, 185)]
# 表頭
header = ['name', 'age', 'height']

with open('person.csv', 'w', encoding='utf-8', newline='') as file_obj:
    # 創建對象
    writer = csv.writer(file_obj)
    # 寫表頭
    writer.writerow(header)
    # 3.寫入數據(一次性寫入多行)
    writer.writerows(person)

寫入結果如下：

第二種寫入方法(使用DictWriter可以使用字典的方式將數據寫入)

注意事項：使用字典的方式寫入要注意傳遞的數據格式必須是字典

如果不是字典的話會報錯

AttributeError: ‘tuple’ object has no attribute ‘keys’

步驟1.創建數據和表頭( 數據必須是字典格式)2.創建DictWriter對象3.寫表頭4.寫入數據

import csv

# 數據
person = [
    {'name': 'xxx', 'age': 18, 'height': 193},
    {'name': 'yyy', 'age': 18, 'height': 182},
    {'name': 'zzz', 'age': 19, 'height': 185},
]
# 表頭
header = ['name', 'age', 'height']

with open('person.csv', 'w', encoding='utf-8', newline='') as file_obj:
    # 1.創建DicetWriter對象
    dictWriter = csv.DictWriter(file_obj, header)
    # 2.寫表頭
    dictWriter.writeheader()
    # 3.寫入數據(一次性寫入多行)
    dictWriter.writerows(person)

csv的讀取

通過reader()讀取

import csv

with open('person.csv', 'r', encoding='utf-8') as file_obj:
    # 1.創建reader對象
    reader = csv.reader(file_obj)
    print(reader)

如果直接打印會返回csv.reader對象，這時需要遍歷列表

<_csv.reader object at 0x000001FB8CE655F8>

改正代碼如下：

import csv

with open('person.csv', 'r', encoding='utf-8') as file_obj:
    # 1.創建reader對象
    reader = csv.reader(file_obj)
    # 2.遍歷進行讀取數據
    for r in reader:
        print(r)

讀取結果如下：

['name', 'age', 'height']
['xxx', '18', '193']
['yyy', '18', '182']
['zzz', '19', '185']

如果想打印列表的某一個值，可以使用索引打印

print(r[0])

name
xxx
yyy
zzz

通過dictreader()讀取

import csv

with open('person.csv', 'r', encoding='utf-8') as file_obj:
    # 1.創建reader對象
    dictReader = csv.DictReader(file_obj)
    # 2.遍歷進行讀取數據
    for r in dictReader:
        print(r)

返回結果如下：

OrderedDict([('name', 'xxx'), ('age', '18'), ('height', '193')])
OrderedDict([('name', 'yyy'), ('age', '18'), ('height', '182')])
OrderedDict([('name', 'zzz'), ('age', '19'), ('height', '185')])

這時我們如果要取到某一個值就需要指定鍵去尋找值

print(r['name'])

xxx
yyy
zzz

以上是“python中csv格式文件如何實現寫入與讀取”這篇文章的所有內容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內容對大家有所幫助，如果還想學習更多知識，歡迎關注億速云行業資訊頻道！