Inflearn Community Q&A

안녕하세요! plotly Heatmap 메서드 옵션에서 숫자를 보여주게 하는거는 못 찾았는데, 레이아웃에 annotation을 추가하면 비슷하게 만들 수 있는 것 같습니다. 혹시라도 도움이 될까봐 코드 공유합니다.(마크업언어가 아니라 코드가 예쁘게 첨부가 안되네요...)

import numpy as np
import plotly.graph_objects as go
np.random.seed(1)

random_matrix = np.random.randint(0, 100, (3, 3))
x = y = ['APPL', 'TSLA', 'AMZ']
trace = go.Heatmap(z=random_matrix,
                   x=['APPL', 'TSLA', 'AMZ'],
                   y=['APPL', 'TSLA', 'AMZ'],)
annotations = go.Annotations()
for r in range(random_matrix.shape[0]):
    for c in range(random_matrix.shape[1]):
        annotations.append(go.Annotation(text=str(random_matrix[r][c]), x=x[c], y=y[r], font={'size': 30, 'color': 'white'}, showarrow=False))
layout = go.Layout(title='Annotated Heatmap', annotations=annotations)
fig = go.Figure(trace, layout)
fig.show()