如何从pandas DataFrame的绘图上绘制带注释的热图?

时间:2019-07-26 10:24:45

标签: plotly plotly-python

我有一个示例DataFrame-df1,如何使用此数据框绘制带注释的热图,以便对所有area_size值进行注释

示例DataFrame(df1):

                     village  crop_name  area_size
0                   Begumpet  SUGARCANE     120.00
1                    Bhadrak  SUGARCANE      10.00
2                 Bowenpally  SUGARCANE      75.00
3                Chandanagar    SOYBEAN       3.00
4                Chandanagar  SUGARCANE      53.00
5              Cheemalamarri     GARLIC       3.00
6              Cheemalamarri     GINGER       5.00
7              Cheemalamarri      ONION       4.00
8              Cheemalamarri     POTATO       4.00
9              Cheemalamarri    SOYBEAN       3.00
10             Cheemalamarri  SUGARCANE       9.00
11             Cheemalamarri   TURMERIC       6.00

我在其他一些stackoverflow链接的帮助下尝试了运气,但没有成功

import plotly.graph_objs as go
x = df1['village']
y = df1['crop_name']
z = df1['area_size']
annotations = go.Annotations()
for n, row in enumerate(z):
    for m, val in enumerate(row):
        annotations.append(go.Annotation(text=str(z[n][m]), x=x[m], y=y[n],
                                         xref='x1', yref='y1', showarrow=False))

colorscale = [[0, '#3D9970'], [1, '#001f3f']]
trace = go.Heatmap(x=x, y=y, z=z, colorscale=colorscale, showscale=False)

fig = go.Figure(data=go.Data([trace]))

预期的输出是一个带注释的热图,因为它是一个交互式的热图,我使用seaborn时不费吹灰之力,也从给定的DataFrame中将其标注为crop_Size的值。

感谢您的帮助,因为我是新手。

0 个答案:

没有答案