Fork me on GitHub

Python 3 爬取城市历史PM2.5数据

发表于 | 阅读量 次
| 字数统计: 751 | 阅读时长 ≈ 3

写在前面：

基于python版本3，使用BeautifulSoup和requests库。
因为近来成都的雾霾越来越严重，影响到的人群也是越来越多，一直想做一个PM2.5的统计，正好最近开始学习Python所有就拿它来练手了，新手不足之处还请指教。

正题：

对比了几个可以查历史数据的网站后，选取的网站是www.aqistudy.cn,此网站的数据易于Python抓取且来源于环保部官方数据，从最早2013年12月到现在的记录，历史数据页面如下图。

图 1 历史数据首页

这次是爬取热门城市里的十个城市的历史数据并保持为CSV格式，分析页面源码可以看到模式如下。

从这个大的<div class="hot">标签里循环每一个 <li> 标签即可以得到每个城市名以及城市的Url。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
### 获取城市名及城市的 URL
def getcity():
    hotcitys = rq.get(Url,headers=headers)
    soup = BeautifulSoup(hotcitys.text, 'lxml')  #这里使用lxml来解析
    citynames = []
    cityurls = []
    citys = soup.find_all('a',limit=19)[-10:]     #分片的形式来获取十个热门城市
    for city in citys:
        cityurl = city.get('href')
        cityurls.append(cityurl)
        cityname = city.get_text()
        citynames.append(cityname)
    return cityurls,citynames
cityurls,citynames = getcity()

导入 python 库

首先导入用到的几个库：

1
2
3
4
import requests as rq
import re
import io
from bs4 import BeautifulSoup

起始网址：Url='https://www.aqistudy.cn/historydata/
为了防止网站针对爬虫的限制，把爬虫伪装成浏览器：
headers = { 'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/55.0.2883.87 Safari/537.36' }

抓取网址

再从抓取到的城市月数据网址开始抓取每月数据的网址：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
def getwebs(url):
    data = rq.get(url,headers=headers).text
    soup = BeautifulSoup(data, 'lxml')
    ahrefs = soup.find_all('a')          #找到所有的链接
    webs = []
    names = []
    for href in ahrefs:
        href = href.get('href')
        hrefd = re.findall('(.*?month=\d{6})',href)  #用正则来匹配每月
        if hrefd:
            urls = Url + hrefd[0]
            webs.append(urls)
    return webs

获得 天气数据

最后把每个城市对应的详细数据写入到CSV文件保存即完成。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
def Savepm25(webs,path):
    for web in webs:
        print('正在下载数据...'+web)
        data = rq.get(web,headers=headers).text
        soup = BeautifulSoup(data, 'lxml')
        pms = soup.find_all('td')
        aqis = []
        for pm in pms:
        
            data = pm.get_text()
            aqis.append(data)
        i = 0
        while i < 12:
            aqis.pop()
            i = i + 1
        with open(path+'.csv', 'a+') as f:
                
             i = 1
             for aqi in aqis:
                if i % 11 != 0:
                    f.write(aqi + ',')       # CSV表格里以','为分隔符
                else:
                    f.write(aqi + '\r\n')    #若是行尾则换行
                i = i + 1

这里with open(path+'.csv', 'a+') as f:，'a+'是以追加的方式添加数据到CSV文件。

保存为 CSV

最后写一个循环来分别以 城市名.CSV 来保存到本地。

1
2
3
4
5
6
7
8
9
10
11
12
paths = []
for cityname in citynames:
    path = cityname + 'pm25'
    paths.append(path)
i = 0
for url in  cityurls:
    url = Url+url
    print(url)
    webs=getwebs(url)
    path = paths[i]
    Savepm25(webs,path)
    i = i + 1

写在最后：

本人刚开始学Python，很多地方没有考虑周到，只实现了基本功能、代码也不够健壮，还请高手可以多指点！或许您的一句话可以让我等新手少走些许弯路！