用Python绘制一个条形图,x轴表示所在部分,y轴表示平均收入,柱子的颜色表明是否离职
时间: 2023-02-13 15:42:48 浏览: 130
为了在 Python 中绘制条形图,你需要安装一个可视化库。Matplotlib 是 Python 中最常用的可视化库之一,它可以轻松地绘制各种图表。
首先,导入 matplotlib 库中的 pyplot 模块。
```
import matplotlib.pyplot as plt
```
然后,你需要准备数据。这里假设你已经有了一个包含所在部分、平均收入和是否离职的数据集。你可以使用列表或 NumPy 数组存储数据。
接下来,使用 plt.bar() 函数绘制条形图。在绘制条形图之前,你还需要调用 plt.xlabel() 和 plt.ylabel() 函数为 x 轴和 y 轴添加标签,使用 plt.title() 函数为图表添加标题。
示例代码如下所示:
```
# 准备数据
department = ['HR', 'R&D', 'Marketing', 'Sales']
salary = [50000, 80000, 45000, 90000]
left = [0, 0, 1, 1] # 0 表示未离职,1 表示已离职
# 绘制条形图
plt.bar(department, salary, color=left)
# 添加标签
plt.xlabel('Department')
plt.ylabel('Salary (USD)')
plt.title('Average Salary by Department')
# 显示图表
plt.show()
```
绘制出来的图表将会显示所在部分与平均收入之间的关系,并且柱子的颜色将表明是否离职。希望这些信息对你有帮助。
阅读全文