Plotly热力图：自定义pandas DataFrame设置表头与索引名称方法

你只需要给自定义DataFrame的行索引、列索引分别设置name属性即可，官方示例数据集的nations就是行索引的名称，medals就是列索引的名称，配置代码如下：

import pandas as pd

# 原有构造DataFrame的逻辑
df = pd.DataFrame(model_data, columns=model_names, index=test_names)

# 新增两行配置，和官方示例属性对齐
df.index.name = "nations"  # 行索引轴名称，可根据你的实际业务场景修改命名
df.columns.name = "medals" # 列索引轴名称，可根据你的实际业务场景修改命名

配置完成后即可直接满足Plotly热力图的调用要求，你可以执行print(df.index.name)、print(df.columns.name)确认配置是否生效。

内容的提问来源于stack exchange，提问作者Jibril