程序员文章、书籍推荐和程序员创业信息与资源分享平台

网站首页 > 技术文章 正文

Python与Excel之间的数据交互(读取、修改、插入和删除数据)

hfteth 2024-12-24 10:32:58 技术文章 36 ℃

引言:

代码对 Excel 文件的多种操作,如读取、修改、插入和删除数据,适用于需要动态操作 Excel 文件内容的场景。其中函数包括读取指定工作表的内容、获取和修改单元格、添加或删除行等,确保所有更改都被保存到文件中。

代码封装如下:

import pandas as pd
from openpyxl import load_workbook
from openpyxl.styles import Border, Side
from openpyxl.utils import get_column_letter

class ExcelHandler:
    #todo初始化ExcelHandler类,加载Excel文件
    def __init__(self, file_path):
        self.file_path = file_path  #todo存储Excel文件的路径
        self.workbook = load_workbook(file_path)  #todo加载工作簿

    #todo获取指定工作表的所有内容
    def get_sheet_content(self, sheet_name):
        rowlist = []
        try:
            sheet = self.workbook[sheet_name]  #todo获取工作表对象
            content = [list(row) for row in sheet.iter_rows(values_only=True)]  #todo获取所有行的内容
            rowlist.extend(content)  #todo将所有行加入到 rowlist
        except Exception as e:
            print(f"获取工作表内容时出错:{e}")
            return None
        return rowlist  #todo返回工作表内容

    #todo获取指定工作表某一列的所有值
    def get_sheet_column_values(self, sheet_name, column_name):
        try:
            df = pd.read_excel(self.file_path, sheet_name=sheet_name, engine='openpyxl', dtype={column_name: str})
            column_values = df[column_name].tolist()  #todo获取该列的所有值
            return column_values
        except Exception as e:
            print(f"获取列值时出错:{e}")
            return None

    #todo获取指定单元格的值
    def get_cell_value(self, sheet_name, cell):
        try:
            sheet = self.workbook[sheet_name]  #todo获取工作表对象
            value = sheet[cell].value  #todo获取单元格的值
            return value
        except Exception as e:
            print(f"获取单元格值时出错:{e}")
            return None

    #todo插入数据到指定单元格(原有值和新值相加)
    def insert_cell_value(self, sheet_name, cell, insert_data):
        try:
            sheet = self.workbook[sheet_name]  #todo获取工作表对象
            current_value = sheet[cell].value  #todo获取当前单元格值
            new_value = current_value + insert_data  #todo新值 = 当前值 + 插入数据
            sheet[cell].value = new_value  #todo设置新值
            self.save_changes()  #todo保存更改
            return True
        except Exception as e:
            print(f"增加单元格值时出错:{e}")
            return False

    #todo删除指定单元格的内容
    def delete_cell_value(self, sheet_name, cell):
        try:
            sheet = self.workbook[sheet_name]  #todo获取工作表对象
            sheet[cell].value = None  #todo设置单元格内容为空
            self.save_changes()  #todo保存更改
            return True
        except Exception as e:
            print(f"删除单元格值时出错:{e}")
            return False

    #todo修改指定单元格的值
    def modify_cell_value(self, sheet_name, cell, new_value):
        try:
            sheet = self.workbook[sheet_name]  #todo获取工作表对象
            sheet[cell].value = new_value  #todo设置单元格的新值
            self.save_changes()  #todo保存更改
            return True
        except Exception as e:
            print(f"修改单元格的值出错:{e}")
            return False

    #todo设置指定单元格的值
    def set_cell_value(self, sheet_name, cell, value):
        try:
            sheet = self.workbook[sheet_name]  #todo获取工作表对象
            sheet[cell].value = value  #todo设置单元格的值
            self.save_changes()  #todo保存更改
            return True
        except Exception as e:
            print(f"设置单元格的值出错:{e}")
            return False

    #todo向指定工作表添加一行数据
    def add_row_to_excel(self, sheet_name, add_data):
        try:
            sheet = self.workbook[sheet_name]  #todo获取工作表对象
            max_row = sheet.max_row  #todo获取当前工作表的最大行数
            for col in range(1, len(add_data) + 1):  #todo遍历每一列
                cell = sheet[get_column_letter(col) + str(max_row + 1)]  #todo获取目标单元格
                cell.value = add_data[col - 1]  #todo设置单元格的值
                #todo设置单元格的边框样式
                border = Border(left=Side(style='thin'), right=Side(style='thin'), 
                                top=Side(style='thin'), bottom=Side(style='thin'))
                cell.border = border
            self.save_changes()  #todo保存更改
            return True
        except Exception as e:
            print(f"新增一行数据时出错:{e}")
            return False

    #todo删除指定工作表的某一行数据
    def delete_row_from_excel(self, sheet_name, row_number):
        try:
            sheet = self.workbook[sheet_name]  #todo获取工作表对象
            sheet.delete_rows(row_number)  #todo删除指定行
            self.save_changes()  #todo保存更改
            return True
        except Exception as e:
            print(f"删除行数据时出错:{e}")
            return False

    #todo保存工作簿的修改
    def save_changes(self):
        try:
            self.workbook.save(self.file_path)  #todo保存工作簿
        except Exception as e:
            print(f"保存文件时出错:{e}")

#todo示例代码,展示如何使用 ExcelHandler 类
if __name__ == '__main__':
    #todo创建 ExcelHandler 对象,传入文件路径
    file_path = r'D:\期货正式员工.xlsx'
    sheet_name = 'all'  #todo指定工作表名称
    column_name = 'CODE'  #todo指定列名称

    #todo创建 ExcelHandler 对象
    excel_handler = ExcelHandler(file_path)
    
    #todo获取指定列的数据
    column_values = excel_handler.get_sheet_column_values(sheet_name, column_name)
    print(column_values)

其中我们可能会遇到如下问题:

文件被其他程序占用:

如果Excel文件在被其他程序(如Excel本身或其他编辑器)占用时,你可能无法进行再次修改。请确保在手动打开文件之前,关闭其他使用该文件的程序。

文件处于只读模式:

如果Excel文件的属性设置为只读模式,你将无法编辑文件。请检查文件的属性,确保访问权限设置正确。

文件被锁定或受保护:

如果Excel文件受到锁定或受保护,你可能需要提供密码或解锁文件才能进行修改。这是为了保护文件内容的安全性。如果你没有正确的密码或权限,你将无法进行编辑。请与文件的创建者或管理员联系以获取进一步的支持。

我用的最简单方法:手动关掉excel文件,然后操作后再重新打开

Tags:

最近发表
标签列表