Matplotlib,Pandas,饼图标签错误

10

我使用Pandas的counts.plot(kind='pie')和Matplotlib的plt.pie(counts)两种方法生成了饼图。问题出在标签上,当我使用自定义颜色和图例时,饼图的标签会偏移。

enter image description here

饼图的标签是正确的,但图例的标签是根据group_name中的标签顺序绘制的,而不是它们的值。有什么解决办法吗?

代码 =

group_names = ['2-3 km', '3-5 km','5-7 km','7-10 km','10-20 km','20-50 km','50-75 km','75-100 km','>100 km']

df['bins'] = pd.cut(df['distkm'], bins)
df['categories'] = pd.cut(df['distkm'], bins, labels=group_names)

counts = df['categories'].value_counts()
plt.axis('equal')
explode = (0, 0, 0,0.1,0.1,0.2,0.3,0.4,0.6)
colors = ['#191970','#001CF0','#0038E2','#0055D4','#0071C6','#008DB8','#00AAAA','#00C69C','#00E28E','#00FF80',]
counts.plot(kind='pie', fontsize=17,colors=colors,explode=explode)
plt.legend(labels=group_names,loc="best")

plt.show()

数据看起来像:

20-50 km     1109
50-75 km      696
10-20 km      353
75-100 km     192
3-5 km        168
7-10 km        86
5-7 km         74
>100 km        65
2-3 km         53
dtype: int64

在“explode”方面,您可以使用“explode=list(accumulate(repeat(.0, len(df["categories"])), lambda a,_: a+0.1))”进行脚本编写。 - A T
3个回答

12

group_names 的顺序与 counts.index 不同。因此,不要使用

plt.legend(labels=group_names,loc="best")

使用

plt.legend(labels=counts.index, loc="best")

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

group_names = ['2-3 km', '3-5 km', '5-7 km', '7-10 km', '10-20 km', '20-50 km', 
               '50-75 km', '75-100 km', '>100 km']

counts = pd.Series([1109, 696, 353, 192, 168, 86, 74, 65, 53], 
                   index=['20-50 km', '50-75 km', '10-20 km', '75-100 km',
                          '3-5 km', '7-10 km', '5-7 km', '>100 km', '2-3 km'])

explode = (0, 0, 0, 0.1, 0.1, 0.2, 0.3, 0.4, 0.6)
colors = ['#191970', '#001CF0', '#0038E2', '#0055D4', '#0071C6', '#008DB8', '#00AAAA',
          '#00C69C', '#00E28E', '#00FF80', ]

counts.plot(kind='pie', fontsize=17, colors=colors, explode=explode)
plt.axis('equal')
plt.ylabel('')
plt.legend(labels=counts.index, loc="best")
plt.show()

在这里输入图像描述


5
如果我取消自动排序,创建值计数,图例将正确绘制。 counts = df ['categories'].value_counts(sort = False)enter image description here

0
您可以将pandas DataFrame转换为列表,并直接使用matplotlib进行绘图。
sizes = counts.values.tolist()
plt.axis('equal')
plt.pie(sizes,explode=explode,labels=group_names,colors=colors)
plt.legend(labels=group_names,loc="best")
plt.show()

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接