帮我改写成利用高德API进行编码的python代码

好的，以下是利用高德API进行编码的 Python 代码：

import requests
import time

# 输入你的高德API Key
key = 'your_key_here'

# 定义函数，传入经度、纬度，返回对应的地址
def get_address(lat, lng):
    # 构造请求URL
    url = 'https://restapi.amap.com/v3/geocode/regeo?key={}&location={:.8f},{:.8f}'.format(key, lng, lat)
    # 发送请求
    res = requests.get(url)
    # 解析结果
    result = res.json()
    if result['status'] == '1':
        return result['regeocode']['formatted_address']
    else:
        return 'unknown'

# 遍历数据集，调用函数进行编码
print("逆地理编码调用API中，需要一定时间...")
for index, row in train.iterrows():
    print("Train - API调用进度：%d/%d" % (index+1, len(train)))
    train.loc[index, 'geo_encode'] = get_address(row['latitude'], row['longitude'])
    time.sleep(0.5) # 等待0.5秒，避免频繁调用API
for index, row in test.iterrows():
    print("Test - API调用进度：%d/%d" % (index+1, len(test)))
    test.loc[index, 'geo_encode'] = get_address(row['latitude'], row['longitude'])
    time.sleep(0.5) # 等待0.5秒，避免频繁调用API

这段代码中，我们使用了 requests 库向高德API发送请求，并解析返回的 JSON 格式数据。在调用 API 时，我们利用了高德API的逆地理编码接口，传入经度和纬度，获取对应的地址信息。在遍历数据集时，我们调用了 get_address() 函数进行编码，并将编码结果保存在数据集的 geo_encode 列中。为了避免频繁调用 API 导致服务拒绝访问，我们使用 time.sleep() 函数进行了等待。