Python | 爬取兴趣点详细攻略（POI）

# -*- coding: utf-8 -*-   
import requests
import json
import osr
import shapefile


class point:
    
    def __init__(self, name, lat,lon):
      self.name = name
      self.lat = lat
      self.lon=lon

def get_json(loc,page_num=0):
    
    header={'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:78.0) Gecko/20100101 Firefox/78.0'}
    
    pa={
        'q':'公园',
        'region':loc,
        'scope':'2',
        'page_size':20,
        'page_num':page_num,
        'output':'json',
        'ak':'替换为你的AK'
        }
    r=requests.get("http://api.map.baidu.com/place/v2/search",params=pa,headers=header)
    my_json=json.loads(r.text)
    return my_json

page=0
last_page=False

info_list=[]
while not last_page:
    park_info=get_json(loc='北京',page_num=page)
    if park_info['results']:
        for each_park in park_info['results']:
            
            try:
                park_name=each_park['name']
            except:
                park_name=None
            try:
                park_lat=each_park['location']['lat']
            except:
                park_lat=None
            try:
                park_lon=each_park['location']['lng']
            except:
                park_lon=None
            
            info_list.append(point(park_name,park_lat,park_lon))
        print('正在爬取第{}页'.format(page+1))
        page=page+1
    else:
        last_page=True

save_path=r'E:\park.shp'
file = shapefile.Writer(save_path,encoding='gbk')
file.field('name', 'C', '40')  
  
for j in info_list:
    if j.lat==None or j.lon==None:
        continue
    else:
        file.point(j.lon,j.lat)
        file.record(j.name)

file.close()

proj = osr.SpatialReference() 
proj.ImportFromEPSG(4326) 
wkt = proj.ExportToWkt()
f = open(save_path.replace(".shp", ".prj"), 'w') 
f.write(wkt)#写入投影信息
f.close()