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 的图标，如果你本身没有安装英文输入法的话是没办法切换的哟。