快捷導航

使用wxPython和pandas模塊生成Excel文件的代碼實現(xiàn)

更新時間：2024年05月11日 11:53:55 作者：winfredzhang

在Python編程中,有時我們需要根據(jù)特定的數(shù)據(jù)生成Excel文件,本文將介紹如何使用wxPython和pandas模塊來實現(xiàn)這個目標,文中通過代碼示例給大家講解的非常詳細,具有一定的參考價值,需要的朋友可以參考下

介紹：

在Python編程中，有時我們需要根據(jù)特定的數(shù)據(jù)生成Excel文件。本文將介紹如何使用wxPython和pandas模塊來實現(xiàn)這個目標。我們將創(chuàng)建一個簡單的GUI應(yīng)用程序，允許用戶選擇輸出文件夾和輸入的Excel文件，并根據(jù)Excel文件中每個單元格的字段名組合生成多個Excel文件。
C:\pythoncode\new\genxlsbyxls.py

1. 準備工作

在開始之前，確保您已經(jīng)安裝了wxPython和pandas模塊。如果沒有安裝，可以使用以下命令進行安裝：

pip install wxPythonpip install pandas

2. 創(chuàng)建GUI窗口

首先，我們需要創(chuàng)建一個GUI窗口，以便用戶可以選擇輸出文件夾和輸入的Excel文件。我們將使用wxPython模塊來創(chuàng)建窗口和按鈕，并將事件與相應(yīng)的回調(diào)函數(shù)關(guān)聯(lián)起來。

import wx
import os
import pandas as pd

class MyFrame(wx.Frame):
    def __init__(self, parent, title):
        super(MyFrame, self).__init__(parent, title=title, size=(400, 200))
        
        panel = wx.Panel(self)
        
        self.output_dir_btn = wx.Button(panel, label="選擇輸出文件夾")
        self.Bind(wx.EVT_BUTTON, self.on_select_output_dir, self.output_dir_btn)
        
        self.input_file_btn = wx.Button(panel, label="選擇Excel文件")
        self.Bind(wx.EVT_BUTTON, self.on_select_input_file, self.input_file_btn)
        
        self.start_btn = wx.Button(panel, label="開始生成")
        self.Bind(wx.EVT_BUTTON, self.on_start_generation, self.start_btn)
        
        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(self.output_dir_btn, 0, wx.ALL|wx.EXPAND, 5)
        sizer.Add(self.input_file_btn, 0, wx.ALL|wx.EXPAND, 5)
        sizer.Add(self.start_btn, 0, wx.ALL|wx.EXPAND, 5)
        panel.SetSizer(sizer)
        
    def on_select_output_dir(self, event):
        dlg = wx.DirDialog(self, "選擇輸出文件夾", style=wx.DD_DEFAULT_STYLE)
        if dlg.ShowModal() == wx.ID_OK:
            self.output_dir = dlg.GetPath()
            print("輸出文件夾:", self.output_dir)
        dlg.Destroy()
    
    def on_select_input_file(self, event):
        dlg = wx.FileDialog(self, "選擇Excel文件", wildcard="Excel files (*.xlsx)|*.xlsx",
                            style=wx.FD_OPEN | wx.FD_FILE_MUST_EXIST)
        if dlg.ShowModal() == wx.ID_OK:
            self.input_file = dlg.GetPath()
            print("輸入文件:", self.input_file)
        dlg.Destroy()
    
    def on_start_generation(self, event):
        if not hasattr(self, 'output_dir') or not hasattr(self, 'input_file'):
            wx.MessageBox("請先選擇輸出文件夾和Excel文件", "錯誤", wx.OK | wx.ICON_ERROR)
            return
        
        df = pd.read_excel(self.input_file)
        
        for i, col in enumerate(df.iloc[:, 0]):
            column_names = col.split(',')
            file_name = f"{i+1}.xlsx"
            file_path = os.path.join(self.output_dir, file_name)
            
            df_new = pd.DataFrame(columns=column_names)
            df_new.to_excel(file_path, index=False)
        
        wx.MessageBox("生成完成", "提示", wx.OK | wx.ICON_INFORMATION)
        self.Close()

app = wx.App()
frame = MyFrame(None, "Excel文件生成器")
frame.Show()
app.MainLoop()```

在上述代碼中，我們創(chuàng)建了一個`MyFrame`類，繼承自wxPython的`Frame`類。該類表示我們的應(yīng)用程序窗口，并包含了選擇輸出文件夾和選擇Excel文件的按鈕。

**3. 實現(xiàn)回調(diào)函數(shù)**

接下來，我們需要實現(xiàn)與按鈕關(guān)聯(lián)的回調(diào)函數(shù)。這些函數(shù)將在用戶點擊相應(yīng)的按鈕時被調(diào)用。

```python
    def on_select_output_dir(self, event):
        dlg = wx.DirDialog(self, "選擇輸出文件夾", style=wx.DD_DEFAULT_STYLE)
        if dlg.ShowModal() == wx.ID_OK:
            self.output_dir = dlg.GetPath()
            print("輸出文件夾:", self.output_dir)
        dlg.Destroy()
    
    def on_select_input_file(self, event):
        dlg = wx.FileDialog(self, "選擇Excel文件", wildcard="Excel files (*.xlsx)|*.xlsx",
                            style=wx.FD_OPEN | wx.FD_FILE_MUST_EXIST)
        if dlg.ShowModal() == wx.ID_OK:
            self.input_file = dlg.GetPath()
            print("輸入文件:", self.input_file)
        dlg.Destroy()
    
    def on_start_generation(self, event):
        if not hasattr(self, 'output_dir') or not hasattr(self, 'input_file'):
            wx.MessageBox("請先選擇輸出文件夾和Excel文件", "錯誤", wx.OK | wx.ICON_ERROR)
            return
        
        df = pd.read_excel(self.input_file)
        
        for i, col in enumerate(df.iloc[:, 0]):
            column_names = col.split(',')
            file_name = f"{i+1}.xlsx"
            file_path = os.path.join(self.output_dir, file_name)
            
            df_new = pd.DataFrame(columns=column_names)
            df_new.to_excel(file_path, index=False)
        
        wx.MessageBox("生成完成", "提示", wx.OK | wx.ICON_INFORMATION)
        self.Close()

在上述代碼中，on_select_output_dir函數(shù)用于選擇輸出文件夾，并將選擇的路徑存儲在self.output_dir變量中。on_select_input_file函數(shù)類似地用于選擇輸入的Excel文件，并將選擇的文件路徑存儲在self.input_file變量中。

on_start_generation函數(shù)是最重要的回調(diào)函數(shù)。它首先讀取選擇的Excel文件，并逐個單元格解析字段名組合。然后，根據(jù)字段名組合創(chuàng)建新的DataFrame，并將其保存為一個新的Excel文件，文件名為序號加上.xlsx后綴。生成的Excel文件將保存在選擇的輸出文件夾中。

4. 運行應(yīng)用程序

最后，我們需要創(chuàng)建一個wx.App對象并運行應(yīng)用程序的主循環(huán)。

app = wx.App()
frame = MyFrame(None, "Excel文件生成器")
frame.Show()
app.MainLoop()

這段代碼創(chuàng)建了一個wx.App對象和一個MyFrame對象，并將應(yīng)用程序的主循環(huán)交給app.MainLoop()處理。

5. 效果演示

在這里插入圖片描述

現(xiàn)在，我們已經(jīng)完成了整個程序的編寫。運行程序后，將會出現(xiàn)一個GUI窗口，您可以通過點擊按鈕選擇輸出文件夾和輸入的Excel文件。當您點擊"開始生成"按鈕后，程序?qū)⒏鶕?jù)Excel文件中的字段名組合生成多個Excel文件，并保存在指定的輸出文件夾中。

這個程序提供了一個簡單而實用的方法來根據(jù)Excel文件中的字段名組合生成多個Excel文件。您可以根據(jù)自己的需求進行修改和擴展，以滿足更復雜的場景。

總結(jié)：

本文介紹了如何使用wxPython和pandas模塊創(chuàng)建一個簡單的GUI應(yīng)用程序，用于根據(jù)Excel文件中的字段名組合生成多個Excel文件。通過選擇輸出文件夾和輸入的Excel文件，程序能夠自動解析字段名組合，并生成對應(yīng)的Excel文件。這種方法可以方便地處理大量數(shù)據(jù)，并將其保存為易于管理和使用的Excel文件。

以上就是使用wxPython和pandas模塊生成Excel文件的代碼實現(xiàn)的詳細內(nèi)容，更多關(guān)于wxPython pandas生成Excel的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: