Pandas与openpyxl库的"完美"融合！

快学Python

发布于 2021-12-08 13:49:58

2K0

发布于 2021-12-08 13:49:58

文章被收录于专栏：快学Python快学Python

大家好，我是黄同学🚀

你用过pandas+openpyxl吗？今天为大家分享一个Python自动化办公文档中，没有提到的知识点。

前言

用过Pandas和openpyxl库的同学都知道，这两个库是相互互补的。Pandas绝对是Python中处理Excel最快、最好用的库，但是使用openpyxl的一些优势是能够轻松地使用样式、条件格式等自定义电子表格。

如果你又想轻松的使用Pandas处理Excel数据，又想为Excel电子表格添加一些样式，应该怎么办呢？

但是您猜怎么着，您不必担心挑选。

事实上，openpyxl 支持将数据从Pandas的DataFrame转换为工作簿，或者相反，将openpyxl工作簿转换为Pandas的DataFrame。

DataFrame转工作簿

我们先创建一个DataFrame：

import pandas as pd

data = {
    "姓名": ["张三", "李四"],
    "性别": ["男", "女"],
    "年龄": [15, 25],
}
df = pd.DataFrame(data)
df

结果如下：

如果想要给表头设置为红色字体，并居中，应该如何设置呢？

from openpyxl import Workbook
from openpyxl.utils.dataframe import dataframe_to_rows
from openpyxl.styles import Font 
from openpyxl.styles import Alignment 

wb = Workbook()
ws = wb.active

for row in dataframe_to_rows(df, index=False, header=True):
    ws.append(row)
    
font = Font(name="微软雅黑",size=10, bold=True,italic=False,color="FF0000")
alignment = Alignment(horizontal="center",vertical="center")
    
for i in range(1,df.shape[1]+1):
    cell = ws.cell(row=1, column=i)
    print(cell.value)
    cell.font = font
    cell.alignment = alignment
    
wb.save("pandas.xlsx")

结果如下：

工作簿转DataFrame

如果有这样一份数据，我们想将其转换为DataFrame，应该怎么做？

其实这个有点多此一举，我们直接使用pandas读取后，处理完数据，在进行样式设计不就行了吗？为何一开始非要使用openpyxl读取工作簿呢？

哈哈，但是既然openpyxl中提供了这种方法，我们就来看看。

import pandas as pd
from openpyxl import load_workbook

wb = load_workbook(filename="df_to_openpyxl.xlsx")
ws = wb.active

values = ws.values
df = pd.DataFrame(values)   
df

结果如下：

本文参与腾讯云自媒体分享计划，分享自微信公众号。

原始发表：2021-12-07，如有侵权请联系 cloudcommunity@tencent.com 删除

python

本文分享自快学Python 微信公众号，前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体分享计划，欢迎热爱写作的你一起参与！

python

登录后参与评论

0 条评论

热度