熊猫数据框样式不适用于Jupyter Notebook

时间:2020-02-05 16:37:53

标签: python r excel pandas dataframe

我有一个数据框,它是使用从Excel文件中读取的列表和其他数据框构造的。 我想做的是,我只需要将背景色应用于要导出到Excel中的数据框的第一行即可。 下面的代码按预期完成了工作。(数据有问题)

问题是我应用于数据框的样式未反映在excel工作表中。我正在使用Jupyter Notebook。 请提出一种在excel中获取样式的方法。

import pandas as pd

sheet1 = r'D:\dinesh\input.xlsx'
sheet2 = "D:\dinesh\lookup.xlsx"
sheet3 = "D:\dinesh\Output.xlsx"
sheetname = 'Dashboard Índice (INPUT)'

print('Started extracting the Crime Type!')
df1 = pd.read_excel(sheet1,sheet_name = 'Dashboard Índice (INPUT)',skiprows=10, usecols = 'B,C,D,F,H,J,L,N,P', encoding = 'unicode_escape')

crime_type = list(df1.iloc[:0])[3:]
print(f'crime_types : {crime_type}')
df1  = (df1.drop(columns=crime_type,axis=1))
cols = list(df1.iloc[0])

print(f'Columns : {cols}')
df1.columns = cols
df1  = (df1[1:]).dropna()

final_data = []
for index, row in df1.iterrows():   
    sheetname   = (f'{row[cols[1]]:0>2d}. {row[cols[0]]}')    
    cnty_cd     = [row[cols[0]], row[cols[1]], row[cols[2]]]
    wb = pd.ExcelFile(sheet2)

    workbook = ''.join([workbook for workbook in wb.sheet_names if workbook.upper() == sheetname])         
    if workbook:
        df2 = pd.read_excel(sheet2, sheet_name = workbook, skiprows=7, usecols ='C,D,H:T', encoding = 'unicode_escape')

        df2_cols = list(df2.columns)  
        final_cols = cols + df2_cols
        df2 = df2.iloc[2:]        
        df2 = df2.dropna(subset=[df2_cols[1]])

        for index2, row2 in df2.iterrows():             
            if row2[df2_cols[1]].upper() in crime_type:
                s1       = pd.Series(cnty_cd)                
                df_rows  = (pd.concat([s1, row2], axis=0)).to_frame().transpose()             
                final_data.append(df_rows)
                break                                                                                 
    else:
        print(f'{sheetname} does not exists!')

df3 = pd.concat(final_data)
df3.columns = final_cols
df_cols     = (pd.Series(final_cols, index=final_cols)).to_frame().transpose()
df_final    = (pd.concat([df_cols,df3], axis=0, ignore_index=True, sort=False))
df_final.style.apply(lambda x: ['background: blue' if x.name==0 else '' for i in x], axis=1)
df_final.to_excel(sheet3, sheet_name='Crime Details',index=False,header = None)

print(f'Sucessfully created the Output file to {sheet3}!')     

1 个答案:

答案 0 :(得分:3)

您需要导出以使样式化的数据框而不是未样式化的数据框更胜一筹,因此您需要链接样式并一起发送到Excel(类似于文档here中所示),或者分配样式化的数据框和用它发送到Excel。
根据您的代码,后者可能看起来像这样:

df_styled = df_final.style.apply(lambda x: ['background: blue' if x.name==0 else '' for i in x], axis=1)
df_styled.to_excel(sheet3,, sheet_name='Crime Details',index=False, header = None, engine='openpyxl')

here所述,您需要使用OpenPyXL或XlsxWriter引擎进行导出。