如何在3列的matplotlib中绘制饼图？

Question

我需要使用matplotlib绘制饼图，但是我的DataFrame有3列，分别是gender，segment和total_amount。 我尝试过使用plt.pie()参数，但是它只使用x和标签作为数据。我尝试将gender设置为图例，但看起来不正确。

import matplotlib.pyplot as plt
import pandas as pd

df = pd.DataFrame({'gender': {0: 'Female',
  1: 'Female',
  2: 'Female',
  3: 'Male',
  4: 'Male',
  5: 'Male'},
 'Segment': {0: 'Gold',
  1: 'Platinum',
  2: 'Silver',
  3: 'Gold',
  4: 'Platinum',
  5: 'Silver'},
 'total_amount': {0: 2110045.0,
  1: 2369722.0,
  2: 1897545.0,
  3: 2655970.0,
  4: 2096445.0,
  5: 2347134.0}})

plt.pie(data = df,x="claim_amount",labels="Segment")
plt.legend(d3.gender)
plt.show()

我想要的结果是一个total_amount的饼图，其标签为gender和segment。如果我能得到该百分比，那将是一个奖励。

Answer 1

我建议以下内容：

# Data to plot
# Take the information from the segment and label columns and join them into one string
labels = df["Segment"]+ " " + df["gender"].map(str)
# Extract the sizes of the segments
sizes = df["total_amount"]
# Plot with labels and percentage
plt.pie(sizes, labels=labels,autopct='%1.1f%%')
plt.show()

您应该获得以下信息：