python 根据 工作年限 学历 和工资绘制折线图
时间: 2023-11-03 19:04:19 浏览: 136
Python折线图绘制
假设有以下数据,表示不同工作年限和学历的平均工资:
```python
data = [
{'exp': 1, 'edu': 'Bachelor', 'salary': 5000},
{'exp': 2, 'edu': 'Bachelor', 'salary': 6000},
{'exp': 3, 'edu': 'Bachelor', 'salary': 7000},
{'exp': 4, 'edu': 'Bachelor', 'salary': 8000},
{'exp': 5, 'edu': 'Bachelor', 'salary': 9000},
{'exp': 1, 'edu': 'Master', 'salary': 6000},
{'exp': 2, 'edu': 'Master', 'salary': 7000},
{'exp': 3, 'edu': 'Master', 'salary': 8000},
{'exp': 4, 'edu': 'Master', 'salary': 9000},
{'exp': 5, 'edu': 'Master', 'salary': 10000},
]
```
您可以使用 Pandas 库进行数据处理和分组,然后使用 Matplotlib 库绘制折线图。
以下是一个简单的例子:
```python
import pandas as pd
import matplotlib.pyplot as plt
# 将数据转换为 DataFrame
df = pd.DataFrame(data)
# 按工作年限和学历分组,并计算平均工资
grouped = df.groupby(['exp', 'edu']).mean()
# 将分组后的数据转换为多级索引的 DataFrame
multi_index = grouped.unstack()
# 绘制折线图
multi_index.plot()
# 添加标题和标签
plt.title('Salary by experience and education')
plt.xlabel('Experience')
plt.ylabel('Salary')
# 显示图表
plt.show()
```
这将绘制一个折线图,其中 x 轴为工作年限,y 轴为平均工资。每条折线代表一种学历。您可以根据需要进行修改和自定义。
阅读全文