为GWR地图创建函数

import geopandas as gpd
import matplotlib.pyplot as plt
from shapely.geometry import Point

def create_gwr_map(gwr_results, geo_data, attribute):
    """
    创建一个GWR结果的地图。
    
    :param gwr_results: GWR模型的结果，应该是一个DataFrame，其中包含每个观测点的参数估计。
    :param geo_data: 包含地理信息的数据框，通常是GeoDataFrame。
    :param attribute: 要在地图上显示的GWR属性列名。
    """
    # 确保geo_data是一个GeoDataFrame
    if not isinstance(geo_data, gpd.GeoDataFrame):
        geo_data = gpd.GeoDataFrame(geo_data)
    
    # 合并GWR结果和地理数据
    merged_data = geo_data.merge(gwr_results[[attribute]], left_index=True, right_index=True)
    
    # 创建地图
    fig, ax = plt.subplots(1, 1, figsize=(10, 10))
    merged_data.plot(column=attribute, cmap='viridis', linewidth=0.8, ax=ax, edgecolor='0.8')
    
    # 添加颜色条
    sm = plt.cm.ScalarMappable(cmap='viridis', norm=plt.Normalize(vmin=merged_data[attribute].min(), vmax=merged_data[attribute].max()))
    sm._A = []
    cbar = fig.colorbar(sm)
    
    # 设置标题
    ax.set_title(f'GWR {attribute} Map')
    
    # 显示地图
    plt.show()

# 假设你已经有了gwr_results和geo_data
# gwr_results = ...
# geo_data = ...

# 调用函数
create_gwr_map(gwr_results, geo_data, 'coefficient')