pandasで表のセルに色を付けたい

Question

#やりたいこと
pythonのpandasで表を作成しています。
そこで、表の中の特定のセルに色を付け、その表を画像データとして保存したい。

#困っていること
以下のサイトを参考にして表を作成していましたが、保存した表に色が付きません。
https://analytics-note.xyz/programming/style-applymap/

現在のコードを以下に示します。
若干コードを諸略していますが、表に関する部分は以下の箇所になります。
また,flow1というリストには、数値データが格納されています。

有識者の方々ご教授のほどよろしく

from argparse import ArgumentParser
import csv
import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns

def cell_style(value):
    if value != value:
        return "background-color: gray; color: white"
    if value <= 40:
        return "background-color: yellow; font-weight: bold"
    else:
        return ""

if __name__ == '__main__':

    list1=["a","b","c","d","e","f"]
    list2=["A","B","C"]
    df=pd.DataFrame(data=flow1, index=list2, columns=list1)

    df.style.applymap(cell_style)

    
    fig,ax = plt.subplots(figsize=((len(df.columns))*1.2, (len(df))*0.5))
    ax.axis('off')
    tbl = ax.table(cellText=df.values,
               bbox=[0,0,1,1],
               colLabels=df.columns,
               rowLabels=df.index)
    plt.savefig('table.png') 

保存される画像

#環境
OS:ubuntu18.04
言語:python2.7

#追記
###プログラム変更点

    
df.style.applymap(cell_style)
   
    colors = [["#56b5fd","w","w"],[ "#1ac3f5","w","w"],[ "#1ac3f5","w","w"]]
    fig,ax = plt.subplots(figsize=((len(df.columns))*1.2, (len(df))*0.5))
    ax.axis('off')
    tbl = ax.table(cellText=df.values,
               bbox=[0,0,1,1],
               colLabels=df.columns,
               rowLabels=df.index,
               cellColours=colors)
    plt.savefig('table.png') 

###エラー

ValueError:Each row in 'cellColours' must have 6 columns

Answer 1

いろいろの資料を調べて、色が付いたDFを画像に直接に転換できないだと思います。
今回は、ax.table(cellColours=colors)　を使って、画像の欄を色が付いた。
このサンプルを参考にしてください。

def cell_style(value):
    if value != value:
        return "background-color: gray; color: white"
    if value <= 40:
        return "background-color: yellow; font-weight: bold"
    else:
        return ""


# 適当なデータフレームを作成
df = pd.DataFrame(
        np.random.randint(0, 100, size=(5, 3)),
        columns=["col0", "col1", "col2"]
    )
df.loc[3, "col0"] = None
df.loc[1, "col2"] = None
df.style.applymap(cell_style)


colors = [["#56b5fd","w","w"],[ "#1ac3f5","w","w"],[ "#1ac3f5","w","w"],[ "#1ac3f5","w","w"],[ "#1ac3f5","w","w"]] 

fig,ax = plt.subplots(figsize=((len(df.columns))*1.2, (len(df))*0.5))
ax.axis('off')
tbl = ax.table(cellText=df.values,
           bbox=[0,0,1,1],
           colLabels=df.columns,
           rowLabels=df.index,
             cellColours=colors )
plt.savefig('table.png') 

Comments

[man_]

FredericChangさん回答ありがとうございます。

私の表は行が3行のため、追記に示すように変更したのですが、エラーが出力されます。エラーの原因は何でしょうか？

[FredericChang]

Before=>colors = [["#56b5fd","w","w"],[ "#1ac3f5","w","w"],[ "#1ac3f5","w","w"]]
After=>colors = [["#56b5fd","w","w","w","w","w"],[ "#1ac3f5","w","w","w","w","w"],[ "#1ac3f5","w","w","w","w","w"]]
colors の中にデータが足りないだと思います。数を六にしてください。

[man_]

ありがとうございます。
表を作成し保存することができました。