python绘制金字塔图代码

在Python中，我们可以使用matplotlib库中的pyplot模块来创建金字塔图。以下是一个简单的例子，展示如何使用`pyplot.bar()`函数创建一个金字塔形数据的图表：

import matplotlib.pyplot as plt

# 定义金字塔的数据
data = [50, 30, 10, 5]

# 创建金字塔图
plt.bar(range(4), data, bottom=0)
plt.xlabel('类别')
plt.ylabel('百分比')
plt.title('金字塔图示例')

# 使用堆叠模式，使得每个条形的高度代表总和
plt.stackplot(range(4), data, labels=['部分A', '部分B', '部分C', '部分D'])
plt.legend(loc='upper left')  # 显示图例
plt.show()

在这个代码中，我们首先导入了matplotlib.pyplot模块，并创建了一个列表`data`存储金字塔各部分的比例。然后通过`bar()`函数创建基本的条形图，`stackplot()`则用于堆积每个部分的数据形成金字塔结构。最后设置了坐标轴标签、标题和图例。