← 返回题库
初级

教育-经验工资热力图(统计图表)

未完成
初级参考 完整示例代码供参考,建议自己理解后重新输入
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())
期望输出
可视化代码执行完成
Python 代码 🔒 登录后使用
🔒

登录后即可练习

注册免费账号,在浏览器中直接运行 Python 代码