我想生成带有一些箭头注释的热图(或更一般地说,在热图上绘制其他视觉对象)。我可以使用matplotlib
来做,但是我想使用plotly
。我知道如何在plotly
中生成热图,但是我不知道如何在其上绘制箭头。
import numpy as np
dimension = 4
Z = np.random.random(pow(dimension, 2)).reshape(dimension, dimension)
这是我的工作matplotlib
示例。
import seaborn as sns
from matplotlib import pyplot as plt
# generate heatmap
sns.heatmap(Z)
for x in range(dimension):
for y in range(dimension):
# draw arrows
plt.arrow(x,y,0.2,0.2,head_width=0.1,color="black",alpha=0.9)
plt.show()
我正在尝试对plotly
做同样的事情,我对库的理解非常有限,所以我知道我可以使用
import plotly.plotly as py
import plotly.graph_objs as go
trace = go.Heatmap(z=Z)
data=[trace]
py.iplot(data, filename='basic-heatmap')
这是来自docs的信息,我可以在线阅读add arrow annotations,但这与我想做的不一样:我不是在寻找注释,而是在寻找一种通用的方式在图形上的plt.arrows
中添加plt.text
或plotly
之类的形状。