logo标签列表

如何设置热力图的宽高比

pythonplotlyheatmapsemantic-segmentation
10

我有一张单通道图像，其中每个整数像素值都映射到一个字符串。例如 5 -> '人'。我想创建一个交互式图片，在悬停在一个像素上时显示它对应的字符串。

我想使用 plotly 热力图来实现这个功能。我遇到的问题如下：

  • 速度非常慢。如果我将 numpy 数组设置为偶数大小（例如 (100,100)），需要花费几分钟才能加载完全。我认为这可能是因为我的代码不够高效？
  • 我无法弄清楚如何保持纵横比例。所以，如果我的图像是一个大小为 (100,100) 的 numpy 数组，我希望绘图也是 (100,100) 像素。
  • 使用 z_text 的空白值似乎是一个糟糕的解决方法，但设置 annotation_text=None 似乎不起作用。

有人可以帮助我吗？以下是我的代码：

import numpy as np
import plotly.graph_objs as go
from plotly.offline import download_plotlyjs, init_notebook_mode, plot, iplot
init_notebook_mode(connected=True)
import plotly.figure_factory as ff

z = np.random.randint(0,6, size=(10, 10))
z_text = np.full(z.shape, '', dtype=str)

d = {0:'a', 1:'b', 2:'c', 3:'d', 4:'e', 5:'f'}
class_mat = np.vectorize(d.get)(z)

fig = ff.create_annotated_heatmap(z, annotation_text=z_text, text=class_mat, hoverinfo='text', colorscale='Viridis', )
fig.layout.title = 'Semantic Segmentation'

iplot(fig, filename='annotated_heatmap_text')

以下是目前的页面显示：

enter image description here

如果Plotly热力图不是最佳选择，我很乐意听取任何替代方案！

注意：我当前正在JupyterLab中显示。

@Austin，我的回答对你有帮助吗？ - vestland
1
虽然我记得是一段时间之前的事情了，但从我的记忆中来看，它对于我的使用情况来说太卡了。尽管如此，我会选择你的答案，因为它似乎在小数据集上运行良好。 - Austin
@Austin 是只有那个卡顿的部分出了问题吗？还是整个文件/笔记本都有问题？ - vestland
如果我没记错的话，只是绘图。这个想法是为至少100x100像素的图像进行操作，所以对于这个来说并不是很有效。也许这只是plotly的限制，不太确定。 - Austin
3个回答
12

我不确定我是否正确了解了每个细节，但是下面片段中的代码将在Jupyter Notebook中生成以下图形。处理纵横比的代码为：

fig['layout']['yaxis']['scaleanchor']='x'

您还可以使用：

fig.update_layout(yaxis = dict(scaleanchor = 'x'))

情节1：

enter image description here

情节2：

请确保包含：

fig.update_layout(plot_bgcolor='rgba(0,0,0,0)')

否则你最终会得到这个：

输入图像描述

代码1 - 我对你的示例进行了编辑：

fig.data[0]['hoverinfo'] = 'all'
fig['layout']['yaxis']['scaleanchor']='x'
fig['layout']['xaxis']['gridcolor'] = 'rgba(0, 0, 0, 0)'
fig['layout']['yaxis']['gridcolor'] = 'rgba(0, 0, 0, 0)'
fig['layout']['yaxis']['color'] = 'rgba(0, 0, 0, 0)'

代码2-整体内容，方便复制粘贴：

import numpy as np
import plotly.graph_objs as go
from plotly.offline import download_plotlyjs, init_notebook_mode, plot, iplot
init_notebook_mode(connected=True)
import plotly.figure_factory as ff

#%qtconsole

z = np.random.randint(0,6, size=(10, 10))
z_text = np.full(z.shape, '', dtype=str)

d = {0:'a', 1:'b', 2:'c', 3:'d', 4:'e', 5:'f'}
class_mat = np.vectorize(d.get)(z)

fig = ff.create_annotated_heatmap(z, annotation_text=z_text,
                                  text=class_mat, hoverinfo='text', colorscale='Viridis',
#                                   x = list('ABCDEFGHIJ'),
#                                   y = list('ABCDEFGHIJ')
                                 )
fig.layout.title = 'Semantic Segmentation'

# My suggestions:
fig.data[0]['hoverinfo'] = 'all'
fig['layout']['yaxis']['scaleanchor']='x'

fig['layout']['xaxis']['gridcolor'] = 'rgba(0, 0, 0, 0)'
fig['layout']['yaxis']['gridcolor'] = 'rgba(0, 0, 0, 0)'
fig['layout']['yaxis']['color'] = 'rgba(0, 0, 0, 0)'

fig.update_layout(plot_bgcolor='rgba(0,0,0,0)')

fig.show()

速度：

即使是这个小图形也需要一些时间来绘制，但目前我还没有任何关于如何加快速度的建议。

1

如果您使用plotly.express.imshow绘制热力图，则该函数有一个参数aspect='auto'，它将更新纵横比以填充绘图所占的空间。

例如：

import plotly.express as px

# fill/load df accordingly to your needs

fig = px.imshow(df, aspect='auto')
0
网页内容由stack overflow 提供, 点击上面的
可以查看英文原文，
原文链接