引言
有朋友发来下面一张图需要代码实现。这种图有水平bar生成,并且每个bar上由两部分叠加组成。接下来就用plotly python这个库来生成这种图。
安装库
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()
最后实现的效果图如下: