Python 之 Excel 表格常用操作

示例文件 test.xlsx

   

将各个表单拆分成单独的 Excel 文件

import os.pathimport openpyxl
import pandasdef handle_excel(file_path):dirname = os.path.dirname(file_path)basename = os.path.basename(file_path).split(".")[0]wb = openpyxl.load_workbook(file_path)sheet_names = wb.sheetnamesprint(sheet_names)for sheet_name in sheet_names:sheet_info = pandas.read_excel(file_path, dtype='str', sheet_name=sheet_name)new_file_path = os.path.join(dirname, f"{basename}_{sheet_name}.xlsx")print(new_file_path)sheet_info.to_excel(new_file_path, index=False)if __name__ == '__main__':file_path = os.path.join(os.getcwd(), "test.xlsx")handle_excel(file_path)

['Sheet1', 'Sheet2']
E:\lky_project\tmp_project\test_project\test_Sheet1.xlsx
E:\lky_project\tmp_project\test_project\test_Sheet2.xlsx

数据分组后按分组结果生成多个 Excel

import os.pathimport openpyxl
import pandasdef handle_excel(file_path):dirname = os.path.dirname(file_path)basename = os.path.basename(file_path).split(".")[0]wb = openpyxl.load_workbook(file_path)sheet_names = wb.sheetnamesprint(sheet_names)sheet_name = sheet_names[1]sheet_info = pandas.read_excel(file_path, dtype='str', sheet_name=sheet_name)group_info = sheet_info.groupby("省份")for key, value in group_info:print(key, value)new_file_path = os.path.join(dirname, f"{basename}_{sheet_name}_{key}.xlsx")print(new_file_path)value.to_excel(new_file_path, index=False)if __name__ == '__main__':file_path = os.path.join(os.getcwd(), "test.xlsx")handle_excel(file_path)

['Sheet1', 'Sheet2']
四川序号  省份 月份  金额
0  1  四川  1  10
1  2  四川  2  20
E:\lky_project\tmp_project\test_project\test_Sheet2_四川.xlsx
陕西序号  省份 月份  金额
2  3  陕西  1  30
E:\lky_project\tmp_project\test_project\test_Sheet2_陕西.xlsx

对 Excel 有效使用区域进行截图保存

import os.path
import xlwings
from PIL import ImageGrab, Image
from xlwings._xlwindows import COMRetryObjectWrapper, App
from win32com.client import DispatchExdef handle_excel(file_path):_xl = COMRetryObjectWrapper(DispatchEx("ket.Application"))impl = App(visible=False, add_book=False, xl=_xl)app = xlwings.App(visible=False, add_book=False, impl=impl)  # 如果运行中不想看到打开 Excel 的操作，设置 visible 为 Falsewb = app.books.open(file_path)sheet = wb.sheets["Sheet1"]all = sheet.used_range  # 获取表格数据使用范围# all = sheet.range("Sheet1!$A$1:$E$4")  # 带表单名称# all = sheet.range("$A$1:$E$4")  # 也可以自定义数据范围，通过对角元素的坐标进行范围限定print(all)all.api.CopyPicture()  # 复制使用范围sheet.api.Paste()  # 粘贴pic = sheet.pictures[0]  # 获取当前图片pic.api.Copy()  # 复制图片到剪切板img = ImageGrab.grabclipboard()  # 获取剪切板的图片数据x, y = img.sizep = Image.new('RGBA', img.size, (255, 255, 255))  # 重新设置背景颜色，不然背景是透明的p.paste(img, (0, 0, x, y), img)  # 将截图粘贴到图片对象p.save("test.png")  # 图片保存pic.delete()  # 删除 sheet 表单粘贴的图片，避免截图影响 Excel 原始数据# wb.save()  # 保存退出wb.close()app.quit()if __name__ == '__main__':file_path = os.path.join(os.getcwd(), "test_Sheet2_四川.xlsx")handle_excel(file_path)

当然，也可以自定义数据截图范围。 

上传文件自动选择

这个和 Excel 没有关系，夹带的私货。

在打开的 windows 窗口自动选择文件并确认

import os.path
import time
import win32con
import win32gui# 自动选择文件并确认
def file_upload(file):retry_times = 3while retry_times > 0:time.sleep(3)dialog = win32gui.FindWindow('#32770', '打开')if dialog:breakretry_times -= 1time.sleep(3)ComboBoxEx32 = win32gui.FindWindowEx(dialog, 0, 'ComboBoxEx32', None)ComboBox = win32gui.FindWindowEx(ComboBoxEx32, 0, 'ComboBox', None)Edit = win32gui.FindWindowEx(ComboBox, 0, 'Edit', None)Button = win32gui.FindWindowEx(dialog, 0, 'Button', None)win32gui.SendMessage(Edit, win32con.WM_SETTEXT, None, file)time.sleep(1)win32gui.SendMessage(dialog, win32con.WM_COMMAND, 1, Button)time.sleep(1)return Trueif __name__ == '__main__':file = os.path.join(os.getcwd(), "test.txt")print(file)file_upload(file)

pip 本地安装依赖包出现 ERROR: No matching distribution found for 报错时：

pip install --no-build-isolation --no-index --find-links=./ fairscale

切换输入法为英文

import win32api
import win32con
import win32guiLANGUAGE = {'ZH': 0x0804,  # 中文（中国）"EN": 0x0409,  # 英语（美国） 美式键盘
}def change_language(language="EN"):language = LANGUAGE.get(language)hwnd = win32gui.GetForegroundWindow()win32api.SendMessage(hwnd, win32con.WM_INPUTLANGCHANGEREQUEST, 0, language)if __name__ == '__main__':change_language('EN')

切换成功以后，右下角的输入法展示会显示一个 ENG 的图标，如果你本身没有安装英文输入法的话是没办法切换的哟。