地理编码Geopandas

pip install geopandas geopy

import geopandas as gpd
import pandas as pd
from geopy.geocoders import Nominatim
from geopy.exc import GeocoderTimedOut

# 创建一个包含地址的 DataFrame
data = {
    'address': [
        '1600 Amphitheatre Parkway, Mountain View, CA',
        '1 Infinite Loop, Cupertino, CA',
        '350 5th Ave, New York, NY'
    ]
}

df = pd.DataFrame(data)

# 初始化地理编码器
geolocator = Nominatim(user_agent="geoapiExercises")

# 定义一个函数来进行地理编码
def geocode_address(address):
    try:
        location = geolocator.geocode(address)
        if location:
            return pd.Series({'latitude': location.latitude, 'longitude': location.longitude})
        else:
            return pd.Series({'latitude': None, 'longitude': None})
    except GeocoderTimedOut:
        return geocode_address(address)  # 递归调用以处理超时

# 应用地理编码函数
df[['latitude', 'longitude']] = df['address'].apply(geocode_address)

# 将 DataFrame 转换为 GeoDataFrame
gdf = gpd.GeoDataFrame(df, geometry=gpd.points_from_xy(df.longitude, df.latitude))

# 显示结果
print(gdf)