日期: 2021 年 4 月 21 日

筛选一个工作簿中的所有工作表数据

import xlwings as xw
import pandas as pd
app = xw.App(visible=False,add_book=False) # 启动excel
workbook = app.books.open(r’d:\python_file\采购表.xlsx’) # 打开要处理的工作簿
worksheet = workbook.sheets # 列出工作簿中的所有工作表
table = pd.DataFrame() # 创建一个空DataFrame
for i, j in enumerate(worksheet): # 遍历工作簿中的工作表
values = j.range(‘a1′).options(pd.DataFrame,header=1,index=False,expand=’table’).value # 读取当前工作表数据
data = values.reindex(columns=[‘采购物品’,’采购日期’,’采购数量’,’采购金额’]) # 调整列的顺序,将”采购物品”移到第1列
table = table.append(data,ignore_index=True) # 将调整列顺序后的数据合并到前面创建的DataFrame中
table = table.groupby(‘采购物品’) # 根据采购物品列筛选数据
new_workbook = xw.books.add() # 新建一个工作簿
for idx, group in table: # 遍历筛选好的数据，其中idx对应物品名称， group对应该物品的所有明细数据
new_worksheet = new_workbook.sheets.add(idx) # 在新工作簿中新增工作表，以物品名称作为工作表名
new_worksheet[‘a1’].options(index=False).value = group # 在新工作表中写入当前物品的所有明细数据
last_cell = new_worksheet[‘a1’].expand(‘table’).last_cell # 获取当前工作表数据区域右下角的单元格
last_row = last_cell.row # 获取数据区域*后一行行号
last_column = last_cell.column # 获取数据区域*后一列的列号
last_column_letter = chr(64 + last_column) # 将数据区域*后一列的列号(数字)转换为该列的列标(字母）
sum_cell_name = ‘{}{}’.format(last_column_letter,last_row+1) # 获取数据区域右下角单元格下方的单元格位置
sum_last_row_name = ‘{}{}’.format(last_column_letter,last_row) # 获取数据区域右下角单元格位置
formula = ‘=SUM({}2:{})’.format(last_column_letter,sum_last_row_name) # 根据前面获取的单元格位置构造excel公式，对采购金额求和
new_worksheet[sum_cell_name].formula = formula # 将求和公式写入数据区域右下解单元格下方的单元格中
new_worksheet.autofit() # 根据单元格中的数据内容自动调整工作表的行高和列宽
new_workbook.save(r’d:\python_file\采购分类表.xlsx’) # 保存新建的工作簿并命名为 “采购分类表.xlsx”
workbook.close()
app.quit()

字节面试中遇到的一道代码题

题目：编写递归算法求*长连续重复的单个字符（如果是找*长重复子串的话，可以参考leetcode 1044: *长重复子串)，比如”aaabbcc”的*大重复字串为”aaa”，“aab”*大重复字串为”aa”, 而类似”aba”这种则返回a,b均可，并编写测试用例。

class Solution:
def repeatTimes(self, s):
n = len(s)
if n == 0:
return None

c = {}
for i in range(n):
if s[i] not in c.keys():
c[s[i]] = 1
else:
c[s[i]] += 1
cmax = max(c.values())
for k, v in c.items():
if v == cmax:
res = k
break
return res * cmax

tests = [”, ‘a’, ‘aa’, ‘abc’, ‘aab’, ‘aba’, ‘aabb’, ‘aabbb’, ‘abcdc’]
a = Solution()
for test in tests:
b = a.repeatTimes(test)
print(b)

批量排序多个工作簿中的数据

import time as tm
t1 = tm.time()
import os
import xlwings as xw
import pandas as pd
app = xw.App(visible=False,add_book=False)
file_path = r’d:\python_file\产品销售统计表’
file_list = os.listdir(file_path)
for i in file_list:
if os.path.splitext(i)[1] == ‘.xlsx’:
workbook = app.books.open(file_path + ‘\\’ + i)
worksheet = workbook.sheets
for j in worksheet:
values = j.range(‘A1’).expand(‘table’).options(pd.DataFrame).value
result = values.sort_values(by=’销售利润’)
j.range(‘A1’).value = result
workbook.save()
workbook.close()
app.quit()
t2 = tm.time()
print(“==========total take {0}”.format(t2 – t1,0.00))