目录
PyQuery 是一个 Python 库,基于 lxml 构建,允许你像 jQuery 一样操作 HTML 文档。它提供了类似
$()
的 API,可以方便地选取和操作 HTML 元素,适用于网页爬取、HTML 解析、数据提取等场景。不需要写正则表达式就可以提取我们想要的东西啦!!!
一、安装依赖库
二、初始化-加载 HTML 文档
PyQuery 支持从字符串、文件或 URL 直接加载 HTML 文档。
(1)字符串初始化-最常用
直接把html的内容当做参数来初始化pyquery对象
from pyquery import PyQuery as pq
html = '''
<div>
<ul>
<li class="item-0">first item</li>
<li class="item-1"><a href="link2.html">second item</a></li>
<li class="item-0 active"><a href="link3.html"><span class="bold">third item</span></a></li>
<li class="item-1 active"><a href="link4.html">fourth item</a></li>
<li class="item-0"><a href="link5.html">fifth item</a></li>
</ul>
</div>
'''
doc = pq(html)
print(doc('li'))
首先引入 PyQuery 这个对象,取别名为 pq。然后声明了一个长 HTML 字符串,并将其当作参数传递给 PyQuery 类,这样就成功完成了初始化。接下来,将初始化的对象传入 CSS 选择器。在这个实例中,我们传入 li 节点,这样就可以选择所有的 li 节点。
(2)url初始化
初始化的参数不仅可以以字符串的形式传递,还可以传入网页的 URL,此时只需要指定参数为 url 即可:
from pyquery import PyQuery as pq
doc = pq(url='http://cuiqingcai.com')
print(doc('title'))
PyQuery 对象会首先请求这个 URL,然后用得到的 HTML 内容完成初始化,这其实就相当于用网页的源代码以字符串的形式传递给 PyQuery 类来初始化。 与下面的功能是相同的:
from pyquery import PyQuery as pq
import requests
doc = pq(requests.get('http://cuiqingcai.com').text)
print(doc('title'))
(3)文件初始化
除了传递一个 URL,还可以传递本地的文件名,参数指定为 filename 即可:
from pyquery import PyQuery as pq
doc = pq(filename='demo.html')
print(doc('li'))
需要有一个本地 HTML 文件 demo.html,其内容是待解析的 HTML 字符串。这样它会首先读取本地的文件内容,然后用文件内容以字符串的形式传递给 PyQuery 类来初始化。
三、pyquery 的基本 CSS 选择器的用法
PyQuery 支持多种 CSS 选择器,包括:
- 标签选择器:
doc('p')
选中所有<p>
标签 - 类选择器:
doc('.info')
选中 class="info" 的元素 - ID 选择器:
doc('#content')
选中 ID 为 content 的元素 - 层级选择器:
doc('#content p')
选中#content
内的所有<p>
元素
我们首先用一个实例来感受一下 pyquery 的 CSS 选择器的用法:
html = '''
<div id="container">
<ul class="list">
<li class="item-0">first item</li>
<li class="item-1"><a href="link2.html">second item</a></li>
<li class="item-0 active"><a href="link3.html"><span class="bold">third item</span></a></li>
<li class="item-1 active"><a href="link4.html">fourth item</a></li>
<li class="item-0"><a href="link5.html">fifth item</a></li>
</ul>
</div>
'''
from pyquery import PyQuery as pq
doc = pq(html)
print(doc('#container .list li'))
print(type(doc('#container .list li')))
初始化 PyQuery 对象之后,传入了一个 CSS 选择器 #container .list li,它的意思是先选取 id 为 container 的节点,然后再选取其内部的 class 为 list 的节点内部的所有 li 节点。然后,打印输出。可以看到,我们成功获取到了符合条件的节点。
最后,将它的类型打印输出。可以看到,它的类型依然是 PyQuery 类型。
如上,我们没有写正则表达式,直接通过选择器和text方法就得到了想要提取的文本信息了
四、常用的查询方法
下面我们介绍一些常用的查询方法,这些函数和 jQuery 中的方法用法也完全相同。
(1)查找子节点
查找子节点需要用到 find 方法,传入的参数是 CSS 选择器,我们还是以上面的 HTML 为例:
doc = pq(html)
items = doc('.list')
print(type(items))
print(items)
lis = items.find('li')
print(type(lis))
print(lis)
选取 class 为 list 的节点,然后调用了 find() 方法,传入 CSS 选择器,选取其内部的 li 节点,最后打印输出。可以发现,find() 方法会将符合条件的所有节点选择出来,结果的类型是 PyQuery 类型。其实 find 的查找范围是节点的所有子孙节点,而如果我们只想查找子节点,那可以用 children 方法:
lis = items.children()
print(type(lis))
print(lis)
如果要筛选所有子节点中符合条件的节点,比如想筛选出子节点中 class 为 active 的节点,可以向 children() 方法传入 CSS 选择器.active:
lis = items.children('.active')
print(lis)
可以看到输出的结果已经做了筛选,留下了 class 为 active 的节点。
(2)父节点
可以用 parent 方法来获取某个节点的父节点,我们用一个实例来感受一下:
html = '''
<div class="wrap">
<div id="container">
<ul class="list">
<li class="item-0">first item</li>
<li class="item-1"><a href="link2.html">second item</a></li>
<li class="item-0 active"><a href="link3.html"><span class="bold">third item</span></a></li>
<li class="item-1 active"><a href="link4.html">fourth item</a></li>
<li class="item-0"><a href="link5.html">fifth item</a></li>
</ul>
</div>
</div>
'''
from pyquery import PyQuery as pq
doc = pq(html)
items = doc('.list')
container = items.parent()
print(type(container))
print(container)
这里我们首先用.list 选取 class 为 list 的节点,然后调用 parent 方法得到其父节点,其类型依然是 PyQuery 类型。这里的父节点是该节点的直接父节点,也就是说,它不会再去查找父节点的父节点,即祖先节点。
输出结果为class = list 的父节点 container
(3)祖先节点
获取某个祖先节点,该怎么办呢?这时可以用 parents 方法:
from pyquery import PyQuery as pq
doc = pq(html)
items = doc('.list')
parents = items.parents()
print(type(parents))
print(parents)
可以看到,输出结果有两个:一个是 class 为 wrap 的节点,一个是 id 为 container 的节点。也就是说,parents() 方法会返回所有的祖先节点。
如果想要筛选某个祖先节点的话,可以向 parents 方法传入 CSS 选择器,这样就会返回祖先节点中符合 CSS 选择器的节点:
parent = items.parents('.wrap')
print(parent)
可以看到,输出结果少了一个节点,只保留了 class 为 wrap 的节点。
(4)兄弟节点
如果要获取兄弟节点,可以使用 siblings() 方法。这里还是以上面的 HTML 代码为例:
from pyquery import PyQuery as pq
doc = pq(html)
li = doc('.list .item-0.active')
print(li.siblings())
这里首先选择 class 为 list 的节点内部 class 为 item-0 和 active 的节点,也就是第三个 li 节点。那么,很明显,它的兄弟节点有 4 个,那就是第一、二、四、五个 li 节点。
运行结果:
如果要筛选某个兄弟节点,我们依然可以向 siblings 方法传入 CSS 选择器,这样就会从所有兄弟节点中挑选出符合条件的节点了:
from pyquery import PyQuery as pq
doc = pq(html)
li = doc('.list .item-0.active')
print(li.siblings('.active'))
五、遍历
对于单个节点来说,可以直接打印输出,也可以直接转成字符串
from pyquery import PyQuery as pq
doc = pq(html)
li = doc('.item-0.active')
print(li)
print(str(li))
输出:
对于多个节点的结果,我们就需要遍历来获取了。例如,这里把每一个 li 节点进行遍历,需要调用 items 方法:
from pyquery import PyQuery as pq
doc = pq(html)
lis = doc('li').items()
print(type(lis))
for li in lis:
print(li, type(li))
运行结果如下:
可以发现,调用 items() 方法后,会得到一个生成器,遍历一下,就可以逐个得到 li 节点对象了,它的类型也是 PyQuery 类型。每个 li 节点还可以调用前面所说的方法进行选择,比如继续查询子节点,寻找某个祖先节点等,非常灵活。
六、获取信息
提取到节点之后,我们的最终目的当然是提取节点所包含的信息了。比较重要的信息有两类,一是获取属性,二是获取文本,下面分别进行说明。
(1)获取属性
提取到某个 PyQuery 类型的节点后,就可以调用 attr() 方法来获取属性:
html = '''
<div class="wrap">
<div id="container">
<ul class="list">
<li class="item-0">first item</li>
<li class="item-1"><a href="link2.html">second item</a></li>
<li class="item-0 active"><a href="link3.html"><span class="bold">third item</span></a></li>
<li class="item-1 active"><a href="link4.html">fourth item</a></li>
<li class="item-0"><a href="link5.html">fifth item</a></li>
</ul>
</div>
</div>
'''
from pyquery import PyQuery as pq
doc = pq(html)
a = doc('.item-0.active a')
print(a, type(a))
print(a.attr('href'))
这里首先选中 class 为 item-0 和 active 的 li 节点内的 a 节点,它的类型是 PyQuery 类型。
然后调用 attr 方法。在这个方法中传入属性的名称,就可以得到这个属性值了。
此外,也可以通过调用 attr 属性来获取属性,用法如下:
print(a.attr.href)
运行结果:
如果选中的是多个元素,然后调用 attr 方法 只会得到第一个节点的属性。 如果想获取所有的 a 节点的属性,就要用到前面所说的遍历了:
from pyquery import PyQuery as pq
doc = pq(html)
a = doc('a')
for item in a.items():
print(item.attr('href'))
(2)获取文本
获取节点之后的另一个主要操作就是获取其内部的文本了,此时可以调用 text 方法来实现:
html = '''
<div class="wrap">
<div id="container">
<ul class="list">
<li class="item-0">first item</li>
<li class="item-1"><a href="link2.html">second item</a></li>
<li class="item-0 active"><a href="link3.html"><span class="bold">third item</span></a></li>
<li class="item-1 active"><a href="link4.html">fourth item</a></li>
<li class="item-0"><a href="link5.html">fifth item</a></li>
</ul>
</div>
</div>
'''
from pyquery import PyQuery as pq
doc = pq(html)
a = doc('.item-0.active a')
print(a)
print(a.text())
这里首先选中一个 a 节点,然后调用 text 方法,就可以获取其内部的文本信息。此时它会忽略掉节点内部包含的所有 HTML,只返回纯文字内容。
但如果想要获取这个节点内部的 HTML 文本,就要用 html 方法了:
from pyquery import PyQuery as pq
doc = pq(html)
li = doc('.item-0.active')
print(li)
print(li.html())
这里我们选中了第三个 li 节点,然后调用了 html() 方法,它返回的结果应该是 li 节点内的所有 HTML 文本。
输出:
如果我们选中的结果是多个节点,html 方法返回的是第一个 li 节点的内部 HTML 文本,而 text 则返回了所有的 li 节点内部的纯文本,中间用一个空格分割开,即返回结果是一个字符串。所以这个地方值得注意,如果得到的结果是多个节点,并且想要获取每个节点的内部 HTML 文本,则需要遍历每个节点。而 text() 方法不需要遍历就可以获取,它将所有节点取文本之后合并成一个字符串。
html = '''
<div class="wrap">
<div id="container">
<ul class="list">
<li class="item-1"><a href="link2.html">second item</a></li>
<li class="item-0 active"><a href="link3.html"><span class="bold">third item</span></a></li>
<li class="item-1 active"><a href="link4.html">fourth item</a></li>
<li class="item-0"><a href="link5.html">fifth item</a></li>
</ul>
</div>
</div>
'''
from pyquery import PyQuery as pq
doc = pq(html)
li = doc('li')
print(li.html())
print(li.text())
print(type(li.text())
七、节点操作
由于节点操作的方法太多,下面举几个典型的例子来说明它的用法。
(1)addClass 和 removeClass
html = '''
<div class="wrap">
<div id="container">
<ul class="list">
<li class="item-0">first item</li>
<li class="item-1"><a href="link2.html">second item</a></li>
<li class="item-0 active"><a href="link3.html"><span class="bold">third item</span></a></li>
<li class="item-1 active"><a href="link4.html">fourth item</a></li>
<li class="item-0"><a href="link5.html">fifth item</a></li>
</ul>
</div>
</div>
'''
from pyquery import PyQuery as pq
doc = pq(html)
li = doc('.item-0.active')
print(li)
li.removeClass('active')
print(li)
li.addClass('active')
print(li)
首先选中了第三个 li 节点,然后调用 removeClass() 方法,将 li 节点的 active 这个 class 移除,后来又调用 addClass() 方法,将 class 添加回来。每执行一次操作,就打印输出当前 li 节点的内容。
可以看到,一共输出了 3 次。第二次输出时,li 节点的 active 这个 class 被移除了,第三次 class 又添加回来了。
所以说,addClass 和 removeClass 方法可以动态改变节点的 class 属性。
(2)attr、text、html
除了操作 class 这个属性外,也可以用 attr 方法对属性进行操作。此外,还可以用 text 和 html 方法来改变节点内部的内容。示例如下:
html = '''
<ul class="list">
<li class="item-0 active"><a href="link3.html"><span class="bold">third item</span></a></li>
</ul>
'''
from pyquery import PyQuery as pq
doc = pq(html)
li = doc('.item-0.active')
print(li)
li.attr('name', 'link')
print(li)
li.text('changed item')
print(li)
li.html('<span>changed item</span>')
print(li)
这里我们首先选中 li 节点,然后调用 attr 方法来修改属性,其中该方法的第一个参数为属性名,第二个参数为属性值。接着,调用 text 和 html 方法来改变节点内部的内容。三次操作后,分别打印输出当前的 li 节点。
所以说,如果 attr 方法只传入第一个参数的属性名,则是获取这个属性值;如果传入第二个参数,可以用来修改属性值。text 和 html 方法如果不传参数,则是获取节点内纯文本和 HTML 文本;如果传入参数,则进行赋值。
(3)remove
html = '''
<div class="wrap">
Hello, World
<p>This is a paragraph.</p>
</div>
'''
from pyquery import PyQuery as pq
doc = pq(html)
wrap = doc('.wrap')
print(wrap.text())
现在想提取 Hello, World 这个字符串,而不要 p 节点内部的字符串,需要怎样操作呢?
wrap.find('p').remove()
print(wrap.text())
首先选中 p 节点,然后调用了 remove() 方法将其移除,然后这时 wrap 内部就只剩下 Hello, World 这句话了,然后再利用 text() 方法提取即可。
另外,其实还有很多节点操作的方法,比如 append()、empty() 和 prepend() 等方法,它们和 jQuery 的用法完全一致,详细的用法可以参考官方文档:pyquery – PyQuery complete API — pyquery 2.0.x documentation
八、伪类选择器
CSS 选择器之所以强大,还有一个很重要的原因,那就是它支持多种多样的伪类选择器,例如选择第一个节点、最后一个节点、奇偶数节点、包含某一文本的节点等。示例如下:
# 导入 PyQuery 库
from pyquery import PyQuery as pq
# 定义 HTML 代码
html = '''
<div class="wrap">
<div id="container">
<ul class="list">
<li class="item-0">first item</li>
<li class="item-1"><a href="link2.html">second item</a></li>
<li class="item-0 active"><a href="link3.html"><span class="bold">third item</span></a></li>
<li class="item-1 active"><a href="link4.html">fourth item</a></li>
<li class="item-0"><a href="link5.html">fifth item</a></li>
</ul>
</div>
</div>
'''
# 解析 HTML
doc = pq(html)
# 选择 <ul> 内的第一个 <li> 元素
li = doc('li:first-child')
print(li) # 输出:<li class="item-0">first item</li>
# 选择 <ul> 内的最后一个 <li> 元素
li = doc('li:last-child')
print(li) # 输出:<li class="item-0"><a href="link5.html">fifth item</a></li>
# 选择 <ul> 内的第二个 <li> 元素(索引从 1 开始)
li = doc('li:nth-child(2)')
print(li) # 输出:<li class="item-1"><a href="link2.html">second item</a></li>
# 选择索引大于 2 的所有 <li> 元素(索引从 0 开始计数,因此 gt(2) 选中索引 3 及之后的元素)
li = doc('li:gt(2)')
print(li)
# 输出:
# <li class="item-1 active"><a href="link4.html">fourth item</a></li>
# <li class="item-0"><a href="link5.html">fifth item</a></li>
# 选择 <ul> 内索引为偶数(2n,即第 2、4、6... 个)的 <li> 元素
li = doc('li:nth-child(2n)')
print(li)
# 输出:
# <li class="item-1"><a href="link2.html">second item</a></li>
# <li class="item-1 active"><a href="link4.html">fourth item</a></li>
# 选择包含文本 "second" 的 <li> 元素
li = doc('li:contains(second)')
print(li) # 输出:<li class="item-1"><a href="link2.html">second item</a></li>
这里我们使用了 CSS3 的伪类选择器,依次选择了第一个 li 节点、最后一个 li 节点、第二个 li 节点、第三个 li 之后的 li 节点、偶数位置的 li 节点、包含 second 文本的 li 节点。
关于 CSS 选择器的更多用法,可以参考 CSS 教程。或者我的这一篇博客 前端【2】html添加样式、CSS选择器-优快云博客
到此为止,pyquery 的常用用法就介绍完了。如果想查看更多的内容,可以参考 pyquery 的官方文档:pyquery: a jquery-like library for python — pyquery 2.0.x documentation。我们相信有了它,解析网页不再是难事。