初级
教育-经验工资热力图(统计图表)
未完成
初级参考
完整示例代码供参考,建议自己理解后重新输入
def solve():
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
url = 'https://liangdaima.com/static/data/wooldridge/wage1.csv'
df = pd.read_csv(url)
df['edu_group'] = pd.cut(df['educ'], bins=[0, 12, 14, 18], labels=['≤12年', '13-14年', '>14年'])
df['exp_group'] = pd.cut(df['exper'], bins=[0, 5, 15, 50], labels=['0-5年', '6-15年', '>15年'])
pivot = df.pivot_table(values='wage', index='exp_group', columns='edu_group', aggfunc='mean')
plt.figure(figsize=(10,6))
plt.imshow(pivot.values, cmap='YlOrRd', aspect='auto')
plt.colorbar(label='平均工资')
plt.xticks(range(len(pivot.columns)), pivot.columns)
plt.yticks(range(len(pivot.index)), pivot.index)
for i in range(len(pivot.index)):
for j in range(len(pivot.columns)):
plt.text(j, i, f'{pivot.values[i,j]:.1f}', ha='center', va='center', fontsize=12)
plt.title('教育-经验 工资热力图', fontsize=14)
plt.xlabel('教育年限', fontsize=12)
plt.ylabel('工作经验', fontsize=12)
plt.tight_layout()
plt.show()
return "可视化代码执行完成"
示例
输入
print(solve())
期望输出
可视化代码执行完成
👑
升级 VIP
解锁全部题目,畅通无阻地学习
- ✓ 解锁全部训练包所有题目
- ✓ 查看完整参考代码和提示
- ✓ 浏览器内直接运行 Python 代码
- ✓ 自动批改 + 进度追踪
30天
¥18
1年
¥99
2年
¥158
3年
¥199