当前位置 博文首页 > 详解Pandas与openpyxl库的超强结合

    详解Pandas与openpyxl库的超强结合

    作者:数据分析与统计学之美 时间:2021-09-14 17:43

    目录
    • 前言
    • DataFrame转工作簿
    • 工作簿转DataFrame

    前言

    用过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
    
    workbook = load_workbook(filename="df_to_openpyxl.xlsx")
    sheet = workbook.active
    
    values = sheet.values
    df = pd.DataFrame(values)   
    df
    

    结果如下:

    在这里插入图片描述

    jsjbwy
    下一篇:没有了