How do I change the order of pie chart slices in Python

user_12126084 0
Viewed 1726

I want to change the order of slices in the pie-chart. Currently, the slices are arranged in descending order. I want the slices of the chart to be in the following order: Yes > Sometimes > Most of the times > no

I am using the following code:

colors = [ '#99f3bd', '#fbaccc','#a8df65', '#ff7b54']
fig, ax = plt.subplots(figsize=(5,15))
ax.set_title('Treatment Group', fontsize=25, fontname="Times New Roman Bold")
ax = df['q6_t'].value_counts(normalize=True).plot.pie(autopct='%1.0f%%', colors = colors)
ax.set_ylabel("")
plt.savefig('q6_t.png', bbox_inches = 'tight', transparent=True)

enter image description here

pandas data-analysis matplotlib python
1 Answers

I am surprised that I have not found a duplicate for this presumably common question. If you want a specific order in the pie plot, you have to sort the pandas series generated by your value counts:

import matplotlib.pyplot as plt
import pandas as pd

#corresponding color-label pairs
colors = ['#99f3bd', '#fbaccc',   '#a8df65',           '#ff7b54']
labels = ["Yes",     "Sometimes", "Most of the times", "No"]

#test data generation
import numpy as np
n=10
np.random.seed(1234)
df=pd.DataFrame({"A": np.random.random(n), "q6_t": np.random.choice(labels, n)})
#print(df)


fig, ax = plt.subplots(figsize=(8, 8))
ax.set_title('Treatment Group', fontsize=25, fontname="Times New Roman Bold")
#reindex(labels) sorts the index of the value counts according to the list labels
ax = df['q6_t'].value_counts(normalize=True).reindex(labels).plot.pie(autopct='%1.0f%%', colors = colors)

ax.set_ylabel("")
plt.show()

Sample output:

enter image description here

user_88811410
Related