房价变量热力图（统计图表）

def solve():
    import pandas as pd
    import matplotlib.pyplot as plt
    import numpy as np
    
    url = 'https://liangdaima.com/static/data/wooldridge/hprice1.csv'
    df = pd.read_csv(url)
    
    corr = df[['price', 'sqrft', 'lotsize', 'bdrms', 'assess']].corr()
    plt.figure(figsize=(8,6))
    plt.imshow(corr, cmap='coolwarm', vmin=-1, vmax=1)
    plt.colorbar(label='相关系数')
    plt.xticks(range(len(corr.columns)), corr.columns, rotation=45)
    plt.yticks(range(len(corr.columns)), corr.columns)
    for i in range(len(corr)):
        for j in range(len(corr)):
            plt.text(j, i, f'{corr.iloc[i,j]:.2f}', ha='center', va='center', fontsize=10)
    plt.title('房价变量相关性热力图', fontsize=14)
    plt.tight_layout()
    plt.show()
    return "可视化代码执行完成"