plotly bar

引言

有朋友发来下面一张图需要代码实现。这种图有水平bar生成,并且每个bar上由两部分叠加组成。接下来就用plotly python这个库来生成这种图。


image.png

安装库

pip install plotly

画图

这种图可参考官方文档[Horizontal Bar Chart] (https://plotly.com/python/horizontal-bar-charts/)
主要是调用plotly的bar接口,并将设置orientation='h'使其变成水平bar。
首先准备数据,并创建一张画布figure,并将认可度和陌生度两种类型的bar添加进figure中,注意这两种bar要设置成堆叠效果,这个会在之后提到,代码如下:


import plotly.graph_objects as go

x_1 = [11, 11, 13, 25, 28, 29, 30, 46, 49, 81]
x_2 = [89, 89, 87, 75, 72, 71, 70, 54, 51, 19]
y = ['上海骏马', '安徽科力', '浙大中控', '连云港杰瑞', '无锡华通', '浙江大华', '易华录', '海康威视', '南京莱斯', '青岛海信']
fig = go.Figure()
fig.add_trace(go.Bar(
    y=y,
    x=x_1,
    name='认可度',
    orientation='h',
    width=0.3,
    marker=dict(
        color='lightblue',
        # line=dict(color='rgba(246, 78, 139, 1.0)', width=3)
    )
))
fig.add_trace(go.Bar(
    y=y,
    x=x_2,
    name='陌生度',
    orientation='h',
    width=0.3,
    marker=dict(
        color='red',
        # line=dict(color='red', width=3)
    )
))

接下来就是添加bar上的百分比标注,可用annotations,在这个例子中需要左右两种不同类型的标注,关于annotations其具体用法参考官方文档,annotations需要设置每个标注的坐标和标注文字,左边的标注一律居中,故横坐标为xd的一半;右边的标注横坐标则为相对应的左边的xd加上本身值的一半,即xd+xd2/2。

annotations = []
for xd, xd2, yd in zip(x_1, x_2, y):
    annotations.append(dict(xref='x', yref='y',
                            x=xd / 2, y=yd,
                            text=str(xd) + '%',
                            font=dict(family='Arial', size=14,
                                      color='black'),
                            showarrow=False))
    annotations.append(dict(xref='x', yref='y',
                                        x=xd + (xd2/2), y=yd,
                                        text=str(xd2) + '%',
                                        font=dict(family='Arial', size=14,
                                                  color='black'),
                                        showarrow=False))

最后设置x轴的标签为百分数,以及在layout中将上面画的两种bar设置成堆叠stack模式,添加标注,以及设置图例legend的位置。具体代码如下:

fig.update_xaxes(
    ticktext=[str(i) + "%" for i in range(0, 101, 10)],
    tickvals=[i for i in range(0, 101, 10)],
)
fig.update_layout(barmode='stack',
                  template="plotly_white",
                  annotations=annotations,
                  legend=dict(x=0.5, y=-0.05, orientation="h"))
fig.show()

最后实现的效果图如下:

image.png

你可能感兴趣的:(plotly bar)