用Python绘制一个条形图,x轴表示所在部分,y轴表示平均收入,柱子的颜色表明是否离职
时间: 2023-02-13 18:42:49 浏览: 38
好的,这是一个简单的教程,希望对您有帮助。
首先,你需要安装 matplotlib 库,这是 Python 中最常用的绘图库之一。你可以使用以下命令安装它:
```python
pip install matplotlib
```
然后,你需要准备你的数据。假设你有如下数据:
|部门|平均收入|是否离职|
|-|-|-|
|销售|10000|否|
|市场|12000|是|
|研发|15000|否|
你可以将它们存储在两个列表中:
```python
departments = ['销售', '市场', '研发']
incomes = [10000, 12000, 15000]
resigned = [False, True, False]
```
现在你已经准备好了数据,可以开始绘图。首先,你需要导入 matplotlib 库并使用它的 `bar` 函数绘制条形图:
```python
import matplotlib.pyplot as plt
plt.bar(departments, incomes)
```
这样就会绘制一个条形图,但是它并不包含离职信息。为了将离职信息包含在图中,你需要给每个柱子设置不同的颜色。你可以使用 matplotlib 的 `color` 参数实现这一点:
```python
colors = ['b' if r else 'g' for r in resigned]
plt.bar(departments, incomes, color=colors)
```
现在你的图中包含了离职信息,但是你可能还希望在图中添加标题、坐标轴标签等元素。你可以使用 mat