Python?使用csv庫處理CSV文件的方法

更新時(shí)間：2023年06月08日 10:03:47 作者：hacker707

Python中集成了專用于處理csv文件的庫，名為：csv，本文給大家介紹了Python使用csv庫處理CSV文件的方法及csv庫中4個(gè)常用的對象，結(jié)合實(shí)例代碼給大家介紹的非常詳細(xì),需要的朋友可以參考下

CSV文件

簡單來說CSV文件由任意數(shù)目的記錄組成，記錄間以某種換行符分隔；每條記錄由字段組成，字段間的分隔符是其它字符或字符串，最常見的是逗號(hào)或制表符。

CSV庫

Python中集成了專用于處理csv文件的庫，名為：csv。

csv 庫中有4個(gè)常用的對象：
csv.reader ：以列表的形式返回讀取的數(shù)據(jù)；
csv.writer ：以列表的形式寫入數(shù)據(jù)；
csv.DictReader ：以字典的形式返回讀取的數(shù)據(jù)；
csv.DictWriter：以字典的形式寫入數(shù)據(jù)；

數(shù)據(jù)處理

原始數(shù)據(jù)

文件名為"test.csv"

在這里插入圖片描述

讀取數(shù)據(jù)

使用csv.reader

# 導(dǎo)入 csv 庫
import csv
def reader():
	# 打開csv文件 此處的delimiter指定了分隔符，不用此參數(shù)的話，分隔符默認(rèn)為逗號(hào)
	with open("test.csv", mode='rt',encoding="utf-8-sig") as f_in:
		reader = csv.reader(f_in,delimiter=',')
		# 獲取第一行的header
	    # header[0] = "name"
	    # header[1] = "sex"
	    # header[2] = "age"
	    # header[3] = "score"
	    header = next(reader)
	    # 逐行獲取數(shù)據(jù)，并處理
		for line in reader:
			print(type(line))

此時(shí)得到的line是一個(gè)list類型

在這里插入圖片描述

使用csv.DictReader

# 導(dǎo)入 csv 庫
import csv
def dict_reader():
	# 打開文件
	with open("test.csv", mode='rt',encoding="utf-8-sig") as f_in:
	    # 基于打開的文件，創(chuàng)建csv.DictReader實(shí)例
	    reader = csv.DictReader(f_in)
	    # 逐行獲取數(shù)據(jù)，并處理
	    for line in reader:
	        print(type(line))

此時(shí)的line就是一個(gè)dict類型

在這里插入圖片描述

寫入數(shù)據(jù)

注意：

在打開文件時(shí)，需要添加newline = “”。否則，會(huì)在每2行有效內(nèi)容之間添加一行空白。
如果要保存的內(nèi)容有中文，而且之后需要用Excel打開文件，那么需要選用utf-8-sig編碼。如果使用utf-8編碼，會(huì)導(dǎo)致使用Excel查看文件時(shí)中文亂碼。

使用csv.writer寫入數(shù)據(jù)

# 導(dǎo)入 csv 庫
import csv
def writer():
	# 創(chuàng)建列表，保存header內(nèi)容
    header_list = ["name", "sex", "age", "score"]
    # 創(chuàng)建列表，保存數(shù)據(jù)
    data_list = [
        ["aaa", "fm", 20, 80],
        ["bbb", "fm", 19, 90],
        ["ccc", "fm", 21, 95]
    ]
    # 以寫方式打開文件。注意添加 newline=""，否則會(huì)在兩行數(shù)據(jù)之間都插入一行空白。
    with open("new_data.csv", mode="w", encoding="utf-8-sig", newline="") as f_out:
        # 基于打開的文件，創(chuàng)建 csv.writer 實(shí)例
        writer = csv.writer(f_out)
        # 寫入 header。
        # writerow() 一次只能寫入一行。
        writer.writerow(header_list)
        # 寫入數(shù)據(jù)。
        # writerows() 一次寫入多行。
        writer.writerows(data_list)

輸出結(jié)果為：

在這里插入圖片描述

使用csv.DictWriter寫入數(shù)據(jù)

# 導(dǎo)入 csv 庫
import csv
def dict_writer():
	# 創(chuàng)建列表，保存header內(nèi)容
    header_list = ["name", "sex", "age", "score"]
    # 創(chuàng)建列表，保存數(shù)據(jù)
    data_list = [
        {"name":"a", "sex":"m", "age":20, "score":80},
        {"name":"bb", "sex":"fm", "age":19, "score":90},
        {"name":"ccc", "sex":"fm", "age":21, "score":95},
    ]
    # 以寫方式打開文件。注意添加 newline=""，否則會(huì)在兩行數(shù)據(jù)之間都插入一行空白。
    with open("dict_new_data.csv", mode="w", encoding="utf-8-sig", newline="") as f_out:
        # 基于打開的文件，創(chuàng)建 csv.DictWriter 實(shí)例，將 header 列表作為參數(shù)傳入。
        writer = csv.DictWriter(f_out, header_list)
        # 寫入 header
        writer.writeheader()
        # 寫入數(shù)據(jù)
        writer.writerows(data_list)

輸出結(jié)果為：

在這里插入圖片描述