从绘图 imshow / 热图的最后一行和最后一列中删除色标

Question

我有一个矩阵，用它来绘制以下图。

正如人们所看到的，绘图的总行和列部分主导了色阶，这反过来又削弱了其他值的视觉效果。我的目标是删除最后一行和最后一列的色阶，但我不确定如何做到这一点。

这是我当前的绘图代码

def annotate_matrix_values(df: pd.DataFrame, fig: go.Figure) -> go.Figure:
    df_matrix = df.to_records(index=False)
    for i, r in enumerate(df_matrix):
        for k, c in enumerate(r):
            fig.add_annotation(
                x=k,
                y=i,
                text=f"<b>{str(int(c))}</b>" if not math.isnan(c) else "",
                showarrow=False,
            )
    fig.update_xaxes(side="top")
    fig.update_coloraxes(showscale=False)
    return fig
  

df_final = pd.crosstab(pt[y_axis_name], pt[x_axis_name], margins=True)

fig = px.imshow(
            df_final,
            labels=dict(x=x_title, y=y_title, color="Value"),
            x=x_label,
            y=y_label,
            color_continuous_scale="Purples",
            aspect="auto",
        )
fig = annotate_matrix_values(df_final, fig)

当然我可以在这里设置 margin false 并得到以下绘图。但我无法显示总数。

df_final = pd.crosstab(pt[y_axis_name], pt[x_axis_name], margins=False)

所以我的目标是显示总数，但上面没有任何色标。感谢您的帮助

Answer 1

了解问题的目的是从色标中排除总数后，我尝试了使用子图并覆盖空白图表和热图的各种方法，但没有得到好的结果。最简单的方法是向 EXPRESS 热图添加四边形和注释。数据以带有分类变量的 x,y 形式创建。热图是使用不包括总计的数据创建的，并且注释是使用总计作为字符串创建的。

import plotly.graph_objects as go
import plotly.express as px
import numpy as np
import pandas as pd

data = np.random.randint(0,10,60).reshape(12,5)

cols = ['Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday']
rows = ['Jan.','Feb.','Mar.','Apr.','May','June','July','Aug.','Sept.','Oct.','Nov.','Dec.']

df = pd.DataFrame(data, columns=cols, index=rows)
df['month_sum'] = df.sum(axis=1)
df.loc['week_sum'] = df.sum(axis=0)

fig = px.imshow(
    df.iloc[:-1,:-1].values,
    labels=dict(x="Day of Week", y="Year of Month", color="Value"),
    x=cols,
    y=rows,
    color_continuous_scale="Purples",
    aspect="auto",
    text_auto='auto'
        )

for i,r in enumerate(df['month_sum']):
    fig.add_shape(type='rect',
                  x0=4.5, y0=-0.5+i, x1=5.5, y1=0.5+i,
                  line=dict(
                      color='rgb(188,189,220)',
                      width=1,
                  ),
                  fillcolor='white',
                 )
    fig.add_annotation(
        x=5,
        y=i,
        text=str(r),
        showarrow=False
    )

for i,c in enumerate(df.loc['week_sum'].tolist()):
    if i == 5:
        break
    fig.add_shape(type='rect',
                  x0=-0.5+i, y0=12.5, x1=0.5+i, y1=11.5,
                  line=dict(
                      color='rgb(188,189,220)',
                      width=1,
                  ),
                  fillcolor='white',
                 )
    fig.add_annotation(
        x=i,
        y=12.0,
        text=str(c),
        showarrow=False
    )

fig.add_annotation(
    x=1.0,
    y=1.04,
    xref='paper',
    yref='paper',
    text='Month Total',
    showarrow=False,
)

fig.add_annotation(
    x=-0.16,
    y=0.02,
    xref='paper',
    yref='paper',
    text='Week Total',
    showarrow=False,
)

fig.update_xaxes(side="top")
fig.update_layout(
    autosize=False,
    width=600,
    height=600,
    coloraxis=dict(showscale=False)
                 )

fig.show()