使用Python分割CSV文件表格

CSV文件是一種常用的數據文件格式，其表格數據可以被多種程序輕鬆讀取和處理。但在實際應用中，我們可能需要根據表格數據的某些特徵將CSV文件分割為多份。本文將介紹如何使用Python實現對CSV文件表格數據的分割。

首先，我們需要讀取CSV文件的表格數據。Python中可以使用csv模塊輕鬆實現這一功能。下面是讀取CSV文件表格數據的代碼：

import csv

with open('data.csv', newline='') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        print(row)

上述代碼使用了Python內置函數open()打開CSV文件，並使用csv.reader()函數讀取文件內的表格數據。在for循環中，我們可以按行逐個讀取文件中的記錄，並通過print()函數輸出到屏幕。

在實際應用中，我們可能需要根據表格數據的行數將CSV文件分割為多個文件。下面是通過設置行數實現表格分割的代碼：

import csv

chunksize = 10000
filename = 'data.csv'

for i, chunk in enumerate(pd.read_csv(filename, chunksize=chunksize)):
    chunk.to_csv('part{}.csv'.format(i), index=False)

上述代碼中，我們使用pandas中的read_csv()函數將整個CSV文件按指定行數分割為多個塊。然後，我們逐個將每個塊保存為單獨的CSV文件。

另外，我們也可以根據CSV文件表格中的某些特徵值進行文件分割。下面是基於特徵值實現表格分割的代碼：

import csv

filename = 'data.csv'

with open(filename) as f:
    reader = csv.DictReader(f)
    writers = {}
    headers = None
    
    for row in reader:
        if headers is None:
            headers = row.keys()
            for header in headers:
                writers[header] = csv.writer(open('{}.csv'.format(header), 'w', newline=''))
                writers[header].writerow(headers)
                
        for k, v in row.items():
            writers[k].writerow([v])

上述代碼中，我們使用了Python內置的csv.DictReader()函數讀取CSV文件，並將表格數據按鍵值對形式存入字典對象中。我們通過構建一個字典writers來實現批量生成與CSV文件表格中的特徵值相關的多個文件。然後，利用csv.writer()函數以逐行的方式向每個文件中寫入表格數據。

分割CSV文件表格是數據處理任務中常見的操作。本文介紹了如何使用Python輕鬆實現表格分割中的基本任務，包括CSV文件表格的讀取和基於特徵值的文件分割。希望本文對你有所幫助。