快捷導(dǎo)航

如何運用python讀寫CSV文件

更新時間：2021年08月18日 17:12:27 作者：TryTestwonderful

本文通過講解使用基礎(chǔ)Python代碼來讀寫和處理CSV文件，點進來的小伙伴一起來學(xué)習(xí)吧

1、使用基礎(chǔ)Python代碼來讀寫和處理CSV文件

import sys
#使用基礎(chǔ)Python代碼來讀寫和處理CSV文件
input_file = sys.argv[1]
output_file = sys.argv[2]
with open(input_file, 'r', newline='') as filereader:
    with open(output_file, 'w', newline='') as filewriter:
        header = filereader.readline()
        header = header.strip()
        header_list = header.split(',')
        print(header_list)
        #map 函數(shù)將 str 函數(shù)應(yīng)用于 header_list 中的每個元素，
        #確保每個元素都是字符串。然后，
        #join 函數(shù)在 header_list 中的每個值之 間插入一個逗號，將這個列表轉(zhuǎn)換為一個字符串
        filewriter.write(','.join(map(str,header_list))+'\n')
        for row in filereader:
            row = row.strip()
            row_list = row.split(',')
            print(row_list)
            filewriter.write(','.join(map(str,row_list))+'\n')

說明：map 函數(shù)將 str 函數(shù)應(yīng)用于 header_list 中的每個元素，確保每個元素都是字符串。然后，
join 函數(shù)在 header_list 中的每個值之間插入一個逗號，將這個列表轉(zhuǎn)換為一個字符串。

執(zhí)行：

2、使用pandas讀寫和處理CSV文件

#使用pandas讀寫和處理CSV文件
import pandas as pd
input_file = sys.argv[1]
output_file = sys.argv[2]
#data_frame：數(shù)據(jù)框也是存儲數(shù)據(jù)的一種方式。
#數(shù)據(jù)框中保留了“表格”這種數(shù)據(jù)組織方式，不需要使用列表套列表的方式來分析數(shù)據(jù)
data_frame = pd.read_csv(input_file)
print(data_frame)
data_frame.to_csv(output_file, index=False)

說明：data_frame：數(shù)據(jù)框也是存儲數(shù)據(jù)的一種方式。數(shù)據(jù)框中保留了“表格”這種數(shù)據(jù)組織方式，不需要使用列表套列表的方式來分析數(shù)據(jù)。

執(zhí)行：

3、使用內(nèi)置csv讀寫和處理CSV文件

#使用內(nèi)置csv讀寫和處理CSV文件，用于正確 處理數(shù)據(jù)值中的嵌入逗號和其他復(fù)雜模式的
import csv
import sys
input_file = sys.argv[1]
output_file = sys.argv[2]
with open(input_file, 'r', newline='') as csv_in_file:
    with open(output_file, 'w', newline='') as csv_out_file:
        #創(chuàng)建文件讀對象
        #第二個參數(shù)（就是 delimiter=','） 是默認分隔符，
        #所以如果你的輸入文件和輸出文件都是用逗號分隔的，就不需要指定這個 參數(shù)
        filereader = csv.reader(csv_in_file, delimiter=',')
        #創(chuàng)建文件寫對象
        filewriter = csv.writer(csv_out_file, delimiter=',')
        for row_list in filereader:
            print(row_list)
            filewriter.writerow(row_list)

說明：第二個參數(shù)（就是 delimiter=','）是默認分隔符，所以如果你的輸入文件和輸出文件都是用逗號分隔的，就不需要指定這個參數(shù)。

執(zhí)行：