当前位置：首页 > article >正文

使用 python中 pandas 将 Excel 转换为 CSV 文件

article 2024/11/1 18:43:27

在数据处理和分析中，我们经常需要将 Excel 文件转换为 CSV 格式。CSV 文件因其简单、易于处理的特点，广泛用于数据交换。本文将介绍如何使用 wxPython 创建一个简单的图形用户界面（GUI），结合 pandas 库，实现 Excel 文件到 CSV 文件的转换功能。
C:\pythoncode\new\excel2csv.py

全部代码

import wx
import pandas as pd
import os

class MyFrame(wx.Frame):
    def __init__(self):
        super().__init__(parent=None, title='Excel to CSV Converter')
        panel = wx.Panel(self)

        self.file_picker = wx.FilePickerCtrl(panel, message="Select an Excel file", wildcard="Excel files (*.xlsx)|*.xlsx")
        self.line_count_label = wx.StaticText(panel, label='Number of rows to convert:')
        self.line_count_text = wx.TextCtrl(panel, value='0')
        self.convert_button = wx.Button(panel, label='Convert to CSV')
        self.convert_button.Bind(wx.EVT_BUTTON, self.on_convert)

        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(self.file_picker, 0, wx.ALL | wx.EXPAND, 5)
        sizer.Add(self.line_count_label, 0, wx.ALL | wx.EXPAND, 5)
        sizer.Add(self.line_count_text, 0, wx.ALL | wx.EXPAND, 5)
        sizer.Add(self.convert_button, 0, wx.ALL | wx.CENTER, 5)

        panel.SetSizer(sizer)
        self.Show()

    def on_convert(self, event):
        xlsx_path = self.file_picker.GetPath()
        if not xlsx_path:
            wx.MessageBox("Please select an Excel file.", "Error", wx.OK | wx.ICON_ERROR)
            return

        # 获取用户输入的行数
        try:
            row_count = int(self.line_count_text.GetValue())
        except ValueError:
            wx.MessageBox("Please enter a valid number.", "Error", wx.OK | wx.ICON_ERROR)
            return

        # 确定输出的CSV文件路径
        csv_path = os.path.splitext(xlsx_path)[0] + '.csv'

        try:
            # 读取Excel文件并转换为CSV，限制转换的行数
            df = pd.read_excel(xlsx_path)
            if row_count > 0:
                df = df.head(row_count)  # 只选择前 row_count 行
            df.to_csv(csv_path, index=False)
            wx.MessageBox(f"Converted to {csv_path}", "Success", wx.OK | wx.ICON_INFORMATION)
        except Exception as e:
            wx.MessageBox(f"Failed to convert: {str(e)}", "Error", wx.OK | wx.ICON_ERROR)

if __name__ == '__main__':
    app = wx.App(False)
    frame = MyFrame()
    app.MainLoop()

环境准备

在开始之前，请确保你已安装以下库：

pip install wxPython pandas openpyxl

wxPython 用于创建 GUI。
pandas 用于处理 Excel 和 CSV 文件。
openpyxl 是 pandas 处理 Excel 文件的依赖。

创建 wxPython 应用程序

我们将创建一个简单的应用程序，允许用户选择一个 Excel 文件，并输入想要转换的行数。以下是完整的代码示例：

import wx
import pandas as pd
import os

class MyFrame(wx.Frame):
    def __init__(self):
        super().__init__(parent=None, title='Excel to CSV Converter')
        panel = wx.Panel(self)

        self.file_picker = wx.FilePickerCtrl(panel, message="Select an Excel file", wildcard="Excel files (*.xlsx)|*.xlsx")
        self.line_count_label = wx.StaticText(panel, label='Number of rows to convert:')
        self.line_count_text = wx.TextCtrl(panel, value='0')
        self.convert_button = wx.Button(panel, label='Convert to CSV')
        self.convert_button.Bind(wx.EVT_BUTTON, self.on_convert)

        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(self.file_picker, 0, wx.ALL | wx.EXPAND, 5)
        sizer.Add(self.line_count_label, 0, wx.ALL | wx.EXPAND, 5)
        sizer.Add(self.line_count_text, 0, wx.ALL | wx.EXPAND, 5)
        sizer.Add(self.convert_button, 0, wx.ALL | wx.CENTER, 5)

        panel.SetSizer(sizer)
        self.Show()

    def on_convert(self, event):
        xlsx_path = self.file_picker.GetPath()
        if not xlsx_path:
            wx.MessageBox("Please select an Excel file.", "Error", wx.OK | wx.ICON_ERROR)
            return

        # 获取用户输入的行数
        try:
            row_count = int(self.line_count_text.GetValue())
        except ValueError:
            wx.MessageBox("Please enter a valid number.", "Error", wx.OK | wx.ICON_ERROR)
            return

        # 确定输出的CSV文件路径
        csv_path = os.path.splitext(xlsx_path)[0] + '.csv'

        try:
            # 读取Excel文件并转换为CSV，限制转换的行数
            df = pd.read_excel(xlsx_path)
            if row_count > 0:
                df = df.head(row_count)  # 只选择前 row_count 行
            df.to_csv(csv_path, index=False)
            wx.MessageBox(f"Converted to {csv_path}", "Success", wx.OK | wx.ICON_INFORMATION)
        except Exception as e:
            wx.MessageBox(f"Failed to convert: {str(e)}", "Error", wx.OK | wx.ICON_ERROR)

if __name__ == '__main__':
    app = wx.App(False)
    frame = MyFrame()
    app.MainLoop()