python腳本實現(xiàn)將excel表格數(shù)據(jù)拆分成兩個文件

更新時間：2025年04月30日 09:50:34 作者：玩人工智能的辣條哥

這篇文章主要為大家詳細介紹了如何用python腳本把一個表格有4萬多條數(shù)據(jù)分為兩個文件表,每個2萬條數(shù)據(jù),文中的示例代碼講解詳細,有需要的小伙伴可以了解下

環(huán)境

python3.10

Win10專業(yè)版

問題描述

如何用python腳本把一個表格有4萬多條數(shù)據(jù)分為兩個文件表，每個2萬條數(shù)據(jù)？

解決方案

你想用 Python 腳本把一個包含 4 萬多條數(shù)據(jù)的表格（假設是 CSV 或 Excel 格式）拆分成兩個文件，每個文件約 2 萬條數(shù)據(jù)。下面給你一個示例腳本，支持 CSV 和 Excel，方便你根據(jù)實際文件格式做調整。

示例代碼（以 CSV 為例）

假設輸入文件是 input.csv，輸出拆分成 output_part1.csv 和 output_part2.csv。

import csv

input_file = 'input.csv'
output_file_1 = 'output_part1.csv'
output_file_2 = 'output_part2.csv'

# 讀取所有數(shù)據(jù)
with open(input_file, newline='', encoding='utf-8') as f_in:
    reader = csv.reader(f_in)
    header = next(reader)  # 讀取表頭
    rows = list(reader)

# 計算拆分點
split_index = 20000

# 分割數(shù)據(jù)
part1 = rows[:split_index]
part2 = rows[split_index:]

# 寫入第一個文件
with open(output_file_1, 'w', newline='', encoding='utf-8') as f_out1:
    writer = csv.writer(f_out1)
    writer.writerow(header)
    writer.writerows(part1)

# 寫入第二個文件
with open(output_file_2, 'w', newline='', encoding='utf-8') as f_out2:
    writer = csv.writer(f_out2)
    writer.writerow(header)
    writer.writerows(part2)

print(f"拆分完成：{len(part1)} 條數(shù)據(jù)寫入 {output_file_1}，{len(part2)} 條寫入 {output_file_2}")

如果是 Excel 文件（例如 .xlsx）

需要先安裝 openpyxl：

pip install openpyxl

示例代碼：

import openpyxl

input_file = 'input.xlsx'
output_file_1 = 'output_part1.xlsx'
output_file_2 = 'output_part2.xlsx'

wb = openpyxl.load_workbook(input_file)
ws = wb.active

# 讀取所有行數(shù)據(jù)（包含表頭）
rows = list(ws.values)
header = rows[0]
data_rows = rows[1:]

split_index = 20000
part1 = data_rows[:split_index]
part2 = data_rows[split_index:]

def save_excel(filename, header, data):
    wb_new = openpyxl.Workbook()
    ws_new = wb_new.active
    ws_new.append(header)
    for row in data:
        ws_new.append(row)
    wb_new.save(filename)

save_excel(output_file_1, header, part1)
save_excel(output_file_2, header, part2)

print(f"拆分完成：{len(part1)} 條數(shù)據(jù)寫入 {output_file_1}，{len(part2)} 條寫入 {output_file_2}")