Plotly 是一个用于创建交互式图表的 Python 库。相关性热图(Correlation Heatmap)是一种用于展示数据集中各个变量之间相关性的图表。颜色标度(Color Scale)用于表示数据的不同值,通常从冷色调(如蓝色)到暖色调(如红色)表示从低到高的值。
Plotly 支持多种类型的热图,包括:
相关性热图常用于以下场景:
要将 Plotly 相关性热图的颜色标度设置为从零(蓝色)到 R(红色),可以使用 colorscale
参数。以下是一个示例代码:
import plotly.express as px
import pandas as pd
# 创建一个示例数据集
data = {
'A': [1, 2, 3, 4],
'B': [4, 3, 2, 1],
'C': [2, 3, 4, 5],
'D': [5, 4, 3, 2]
}
df = pd.DataFrame(data)
# 计算相关性矩阵
corr_matrix = df.corr()
# 创建热图
fig = px.imshow(corr_matrix,
labels=dict(x="Features", y="Features", color="Correlation"),
x=corr_matrix.columns,
y=corr_matrix.columns,
color_continuous_scale=[[0, 'blue'], [1, 'red']])
# 显示图表
fig.show()
px.imshow
函数创建热图,并通过 color_continuous_scale
参数设置颜色标度。fig.show()
显示图表。通过上述步骤,你可以将 Plotly 相关性热图的颜色标度设置为从零(蓝色)到 R(红色)。
领取专属 10元无门槛券
手把手带您无忧上云