快捷導(dǎo)航

python使用pandas實現(xiàn)Excel轉(zhuǎn)換為CSV文件

更新時間：2024年10月28日 09:54:51 作者：winfredzhang

在數(shù)據(jù)處理和分析中,我們經(jīng)常需要將 Excel 文件轉(zhuǎn)換為 CSV 格式,這篇文章我們主要來介紹一下python如何使用pandas實現(xiàn)Excel轉(zhuǎn)換為CSV文件,希望對大家有所幫助

在數(shù)據(jù)處理和分析中，我們經(jīng)常需要將 Excel 文件轉(zhuǎn)換為 CSV 格式。CSV 文件因其簡單、易于處理的特點，廣泛用于數(shù)據(jù)交換。本文將介紹如何使用 wxPython 創(chuàng)建一個簡單的圖形用戶界面（GUI），結(jié)合 pandas 庫，實現(xiàn) Excel 文件到 CSV 文件的轉(zhuǎn)換功能。

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

        # 獲取用戶輸入的行數(shù)
        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文件并轉(zhuǎn)換為CSV，限制轉(zhuǎn)換的行數(shù)
            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()

環(huán)境準(zhǔn)備

在開始之前，請確保你已安裝以下庫：

pip install wxPython pandas openpyxl

wxPython 用于創(chuàng)建 GUI。
pandas 用于處理 Excel 和 CSV 文件。
openpyxl 是 pandas 處理 Excel 文件的依賴。

創(chuàng)建 wxPython 應(yīng)用程序

我們將創(chuàng)建一個簡單的應(yīng)用程序，允許用戶選擇一個 Excel 文件，并輸入想要轉(zhuǎn)換的行數(shù)。以下是完整的代碼示例：

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

        # 獲取用戶輸入的行數(shù)
        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文件并轉(zhuǎn)換為CSV，限制轉(zhuǎn)換的行數(shù)
            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()