python提取特定格式的數據的操作方法

更新時間：2024年06月26日 09:00:26 作者：M J S

這篇文章主要介紹了python提取特定格式的數據,通過本腳本,我們學習了如何使用 Pandas 讀取和處理 Excel 數據,如何使用 Tkinter 創(chuàng)建圖形用戶界面,以及如何處理文件對話框和文件操作,需要的朋友可以參考下

導入庫

腳本使用了以下主要庫：

tkinter：用于創(chuàng)建圖形用戶界面。
pandas：用于處理 Excel 數據。
os：用于處理文件和目錄路徑。

import tkinter as tk
from tkinter import filedialog, messagebox
import pandas as pd
import os

Pandas 數據處理

讀取 Excel 文件

使用 pd.read_excel 方法讀取 Excel 文件，并使用 sheet_name=None 參數讀取所有工作表。添加 index_col=None 參數以確保第一列不會被自動設置為索引列。

source_df = pd.read_excel(file_path, sheet_name=None, index_col=None)
source_data = source_df['一格一案']

數據提取

通過 Pandas 的 iloc 方法，根據行列索引提取特定數據。

result_data = {
    '網格編號': source_data.iloc[1, 1],
    '責任段': source_data.iloc[1, 3],
    ...
}

處理合并單元格數據：

risk_check_path = "\n".join(source_data.iloc[9:19, 1].dropna().astype(str))
result_data['五、風險項點檢查路徑'] = risk_check_path

創(chuàng)建 DataFrame 并導出為 Excel 文件

將所有提取的數據放入一個 DataFrame 中，并使用 to_excel 方法導出為 Excel 文件。

result_df = pd.DataFrame(all_data)
result_df.to_excel(output_file_path, index=False)

Tkinter GUI 界面

創(chuàng)建主窗口

使用 tk.Tk 創(chuàng)建主窗口，并設置窗口標題、大小和位置。

root = tk.Tk()
root.title("Excel 轉換工具")
root.geometry(f'{window_width}x{window_height}+{position_right}+{position_top}')

創(chuàng)建按鈕和標簽

使用 tk.Button 和 tk.Label 創(chuàng)建按鈕和標簽，并設置其屬性和布局。

title_label = tk.Label(root, text="Excel 轉換工具", font=("Arial", 18))
title_label.pack(pady=20)
select_button = tk.Button(root, text="選擇 Excel 文件", command=select_files, font=("Arial", 12))
select_button.pack(pady=10)

文件操作

文件對話框

使用 filedialog.askopenfilenames 打開文件選擇對話框，允許用戶選擇多個 Excel 文件。使用 filedialog.asksaveasfilename 打開文件保存對話框，允許用戶選擇保存路徑。

file_paths = filedialog.askopenfilenames(filetypes=[("Excel 文件", "*.xlsx")])
output_file_path = filedialog.asksaveasfilename(defaultextension=".xlsx", filetypes=[("Excel 文件", "*.xlsx")])

主要函數解釋

`transform_to_result_format_specific`

該函數從源數據中提取特定字段，并返回一個字典格式的結果數據。

def transform_to_result_format_specific(source_data, source_file_path):
    risk_check_path = "\n".join(source_data.iloc[9:19, 1].dropna().astype(str))
    result_data = { ... }
    return result_data

`select_files`

該函數處理文件選擇、數據轉換和結果保存的主要邏輯。

def select_files():
    file_paths = filedialog.askopenfilenames(filetypes=[("Excel 文件", "*.xlsx")])
    all_data = []
    for file_path in file_paths:
        source_df = pd.read_excel(file_path, sheet_name=None, index_col=None)
        source_data = source_df['一格一案']
        transformed_data = transform_to_result_format_specific(source_data, file_path)
        all_data.append(transformed_data)
    result_df = pd.DataFrame(all_data)
    output_file_path = filedialog.asksaveasfilename(defaultextension=".xlsx", filetypes=[("Excel 文件", "*.xlsx")])
    if output_file_path:
        result_df.to_excel(output_file_path, index=False)
        messagebox.showinfo("成功", "文件已成功轉換并保存。")