绘制一个热力图,显示每个部门与员工工作年限和工资之间的关系。 要求: • 横轴表示 "Department"(部门),纵轴表示 "Years_of_work"(工作年限),每个方格的颜色深浅代表工资的高低。
时间: 2024-09-23 14:15:03 浏览: 312
一个绘制混淆矩阵热力图的示例代码
绘制热力图(也称为热度图或热映射)通常使用数据可视化工具,如Python的seaborn、matplotlib或者Tableau等。这里是一个简单的例子,使用pandas、numpy和seaborn库在Python中创建一个这样的图:
```python
import pandas as pd
import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt
# 假设你有一个名为df的DataFrame,其中包含以下列:'Department', 'Years_of_work', 'Salary'
# df = ...
# 将 Salary 列转换为数值,方便颜色映射
df['Salary'] = df['Salary'].astype(float)
# 创建热力图矩阵
heatmap_data = df.pivot('Department', 'Years_of_work', 'Salary')
# 设置颜色映射,比如使用蓝到红的渐变
cmap = sns.diverging_palette(230, 20, n=8, center="light", as_cmap=True)
# 创建热力图
sns.heatmap(heatmap_data, cmap=cmap, annot=True, fmt=".2f") # 显示每个方格的具体值
# 设置坐标轴标签和标题
plt.xlabel('Department')
plt.ylabel('Years of Work')
plt.title('Heatmap showing salary vs department and work experience')
# 显示图形
plt.show()
```
在这个示例中,颜色越深表示工资越高,淡色则表示工资较低。你可以根据实际数据调整颜色映射和刻度。
阅读全文