pandas style设置渐变背景色 background_gradient

渐变背景色可以通过设置.background_gradient()实现

更多渐变色请参考：https://matplotlib.org/stable/users/explain/colors/colormaps.html

一个简单的案例：

import pandas as pd
import numpy as np
import dataframe_image as dfi


def send_pic(excel_df, fig_title):
    df_style = excel_df.style.background_gradient(cmap="YlGn_r", subset=['col_1'])
    dfi.export(obj=df_style, filename=f"{fig_title}.png", fontsize=30) # 把dataframe变成图片


if __name__ == '__main__':
    dataframe = pd.DataFrame(np.random.random(size=(10, 5)))
    dataframe.columns = ['col_1', "col_2", "col_3", "col_4", "col_5"]
    send_pic(dataframe, "渐变色")

效果图如下所示：

完整示例程序

import pandas as pd
import numpy as np
import dataframe_image as dfi


def send_pic(excel_df, fig_title):
    df_style = excel_df.style.set_precision(3) \
        .set_properties(**{'text-align': 'center',
                           'border-bottom-color': 'black',
                           'border-bottom-width': '2px',
                           'border-bottom-style': 'groove'}) \
        .hide_index().background_gradient(cmap="YlGn_r", subset=['col_1'])
    dfi.export(obj=df_style, filename=f"{fig_title}.png", fontsize=30)


if __name__ == '__main__':
    dataframe = pd.DataFrame(np.random.random(size=(10, 5)))
    dataframe.columns = ['col_1', "col_2", "col_3", "col_4", "col_5"]
    send_pic(dataframe, "渐变色")

最终效果：