本節中,我們看一下正規表示式的相關用法。正規表示式是處理字串的強大工具,它有自己特定的語法結構,有了它,實現字串的檢索、替換、匹配驗證都不在話下。
當然,對於爬蟲來說,有了它,從HTML裡提取想要的資訊就非常方便了。
1. 例項引入
說了這麼多,可能我們對它到底是個什麼還是比較模糊,下面就用幾個例項來看一下正規表示式的用法。
開啟開源中國提供的正規表示式測試工具tool.oschina.net/regex/,輸入待匹配的文字,然後選擇常用的正規表示式,就可以得出相應的匹配結果了。例如,這裡輸入待匹配的文字如下:
Hello, my phone number is 010-86432100 and email is cqc@cuiqingcai.com, and my website is http://cuiqingcai.com.
複製程式碼
這段字串中包含了一個電話號碼和一個電子郵件,接下來就嘗試用正規表示式提取出來,如圖3-10所示。
圖3-10 執行頁面
在網頁右側選擇“匹配Email地址”,就可以看到下方出現了文字中的E-mail。如果選擇“匹配網址URL”,就可以看到下方出現了文字中的URL。是不是非常神奇?
其實,這裡就是用了正規表示式匹配,也就是用一定的規則將特定的文字提取出來。比如,電子郵件開頭是一段字串,然後是一個@符號,最後是某個域名,這是有特定的組成格式的。另外,對於URL,開頭是協議型別,然後是冒號加雙斜線,最後是域名加路徑。
對於URL來說,可以用下面的正規表示式匹配:
[a-zA-z]+://[^\s]*
複製程式碼
用這個正規表示式去匹配一個字串,如果這個字串中包含類似URL的文字,那就會被提取出來。
這個正規表示式看上去是亂糟糟的一團,其實不然,這裡面都是有特定的語法規則的。比如,a-z
代表匹配任意的小寫字母,\s
表示匹配任意的空白字元,*
就代表匹配前面的字元任意多個,這一長串的正規表示式就是這麼多匹配規則的組合。
寫好正規表示式後,就可以拿它去一個長字串裡匹配查詢了。不論這個字串裡面有什麼,只要符合我們寫的規則,統統可以找出來。對於網頁來說,如果想找出網頁原始碼裡有多少URL,用匹配URL的正規表示式去匹配即可。
上面我們說了幾個匹配規則,表3-2列出了常用的匹配規則。
表3-2 常用的匹配規則
看完了之後,可能有點暈暈的吧,不過不用擔心,後面我們會詳細講解一些常見規則的用法。
其實正規表示式不是Python獨有的,它也可以用在其他程式語言中。但是Python的re庫提供了整個正規表示式的實現,利用這個庫,可以在Python中使用正規表示式。在Python中寫正規表示式幾乎都用這個庫,下面就來了解它的一些常用方法。
2. match()
這裡首先介紹第一個常用的匹配方法——match()
,向它傳入要匹配的字串以及正規表示式,就可以檢測這個正規表示式是否匹配字串。
match()
方法會嘗試從字串的起始位置匹配正規表示式,如果匹配,就返回匹配成功的結果;如果不匹配,就返回None
。示例如下:
import re
content = 'Hello 123 4567 World_This is a Regex Demo'
print(len(content))
result = re.match('^Hello\s\d\d\d\s\d{4}\s\w{10}', content)
print(result)
print(result.group())
print(result.span())
複製程式碼
執行結果如下:
41
<_sre.SRE_Match object; span=(0, 25), match='Hello 123 4567 World_This'>
Hello 123 4567 World_This
(0, 25)
複製程式碼
這裡首先宣告瞭一個字串,其中包含英文字母、空白字元、數字等。接下來,我們寫一個正規表示式:
^Hello\s\d\d\d\s\d{4}\s\w{10}
複製程式碼
用它來匹配這個長字串。開頭的^
是匹配字串的開頭,也就是以Hello
開頭;然後\s
匹配空白字元,用來匹配目標字串的空格;\d
匹配數字,3個\d
匹配123;然後再寫1個\s
匹配空格;後面還有4567
,我們其實可以依然用4個\d
來匹配,但是這麼寫比較煩瑣,所以後面可以跟{4}
以代表匹配前面的規則4次,也就是匹配4個數字;然後後面再緊接1個空白字元,最後\w{10}
匹配10個字母及下劃線。我們注意到,這裡其實並沒有把目標字串匹配完,不過這樣依然可以進行匹配,只不過匹配結果短一點而已。
而在match()
方法中,第一個引數傳入了正規表示式,第二個引數傳入了要匹配的字串。
列印輸出結果,可以看到結果是SRE_Match
物件,這證明成功匹配。該物件有兩個方法:group()
方法可以輸出匹配到的內容,結果是Hello 123 4567 World_This
,這恰好是正規表示式規則所匹配的內容;span()
方法可以輸出匹配的範圍,結果是(0, 25)
,這就是匹配到的結果字串在原字串中的位置範圍。
通過上面的例子,我們基本瞭解瞭如何在Python中使用正規表示式來匹配一段文字。
匹配目標
剛才我們用match()
方法可以得到匹配到的字串內容,但是如果想從字串中提取一部分內容,該怎麼辦呢?就像最前面的例項一樣,從一段文字中提取出郵件或電話號碼等內容。
這裡可以使用()
括號將想提取的子字串括起來。()
實際上標記了一個子表示式的開始和結束位置,被標記的每個子表示式會依次對應每一個分組,呼叫group()
方法傳入分組的索引即可獲取提取的結果。示例如下:
import re
content = 'Hello 1234567 World_This is a Regex Demo'
result = re.match('^Hello\s(\d+)\sWorld', content)
print(result)
print(result.group())
print(result.group(1))
print(result.span())
複製程式碼
這裡我們想把字串中的1234567
提取出來,此時可以將數字部分的正規表示式用()
括起來,然後呼叫了group(1)
獲取匹配結果。
執行結果如下:
<_sre.SRE_Match object; span=(0, 19), match='Hello 1234567 World'>
Hello 1234567 World
1234567
(0, 19)
複製程式碼
可以看到,我們成功得到了1234567
。這裡用的是group(1)
,它與group()
有所不同,後者會輸出完整的匹配結果,而前者會輸出第一個被()
包圍的匹配結果。假如正規表示式後面還有()
包括的內容,那麼可以依次用group(2)
、group(3)
等來獲取。
通用匹配
剛才我們寫的正規表示式其實比較複雜,出現空白字元我們就寫\s
匹配,出現數字我們就用\d
匹配,這樣的工作量非常大。其實完全沒必要這麼做,因為還有一個萬能匹配可以用,那就是.*
(點星)。其中.
(點)可以匹配任意字元(除換行符),*
(星)代表匹配前面的字元無限次,所以它們組合在一起就可以匹配任意字元了。有了它,我們就不用挨個字元地匹配了。
接著上面的例子,我們可以改寫一下正規表示式:
import re
content = 'Hello 123 4567 World_This is a Regex Demo'
result = re.match('^Hello.*Demo$', content)
print(result)
print(result.group())
print(result.span())
複製程式碼
這裡我們將中間部分直接省略,全部用.*
來代替,最後加一個結尾字串就好了。執行結果如下:
<_sre.SRE_Match object; span=(0, 41), match='Hello 123 4567 World_This is a Regex Demo'>
Hello 123 4567 World_This is a Regex Demo
(0, 41)
複製程式碼
可以看到,group()
方法輸出了匹配的全部字串,也就是說我們寫的正規表示式匹配到了目標字串的全部內容;span()
方法輸出(0, 41)
,這是整個字串的長度。
因此,我們可以使用.*
簡化正規表示式的書寫。
貪婪與非貪婪
使用上面的通用匹配.*
時,可能有時候匹配到的並不是我們想要的結果。看下面的例子:
import re
content = 'Hello 1234567 World_This is a Regex Demo'
result = re.match('^He.*(\d+).*Demo$', content)
print(result)
print(result.group(1))
複製程式碼
這裡我們依然想獲取中間的數字,所以中間依然寫的是(\d+)
。而數字兩側由於內容比較雜亂,所以想省略來寫,都寫成 .*
。最後,組成^He.*(\d+).*Demo$
,看樣子並沒有什麼問題。我們看下執行結果:
<_sre.SRE_Match object; span=(0, 40), match='Hello 1234567 World_This is a Regex Demo'>
7
複製程式碼
奇怪的事情發生了,我們只得到了7這個數字,這是怎麼回事呢?
這裡就涉及一個貪婪匹配與非貪婪匹配的問題了。在貪婪匹配下,.*
會匹配儘可能多的字元。正規表示式中.*
後面是\d+
,也就是至少一個數字,並沒有指定具體多少個數字,因此,.*
就儘可能匹配多的字元,這裡就把123456
匹配了,給\d+
留下一個可滿足條件的數字7,最後得到的內容就只有數字7了。
但這很明顯會給我們帶來很大的不便。有時候,匹配結果會莫名其妙少了一部分內容。其實,這裡只需要使用非貪婪匹配就好了。非貪婪匹配的寫法是.*?
,多了一個?
,那麼它可以達到怎樣的效果?我們再用例項看一下:
import re
content = 'Hello 1234567 World_This is a Regex Demo'
result = re.match('^He.*?(\d+).*Demo$', content)
print(result)
print(result.group(1))
複製程式碼
這裡我們只是將第一個.*
改成了.*?
,轉變為非貪婪匹配。結果如下:
<_sre.SRE_Match object; span=(0, 40), match='Hello 1234567 World_This is a Regex Demo'>
1234567
複製程式碼
此時就可以成功獲取1234567
了。原因可想而知,貪婪匹配是儘可能匹配多的字元,非貪婪匹配就是儘可能匹配少的字元。當.*?
匹配到Hello
後面的空白字元時,再往後的字元就是數字了,而\d+
恰好可以匹配,那麼這裡.*?
就不再進行匹配,交給\d+
去匹配後面的數字。所以這樣.*?
匹配了儘可能少的字元,\d+
的結果就是1234567
了。
所以說,在做匹配的時候,字串中間儘量使用非貪婪匹配,也就是用.*?
來代替.*
,以免出現匹配結果缺失的情況。
但這裡需要注意,如果匹配的結果在字串結尾,.*?
就有可能匹配不到任何內容了,因為它會匹配儘可能少的字元。例如:
import re
content = 'http://weibo.com/comment/kEraCN'
result1 = re.match('http.*?comment/(.*?)', content)
result2 = re.match('http.*?comment/(.*)', content)
print('result1', result1.group(1))
print('result2', result2.group(1))
複製程式碼
執行結果如下:
result1
result2 kEraCN
複製程式碼
可以觀察到,.*?
沒有匹配到任何結果,而.*
則儘量匹配多的內容,成功得到了匹配結果。
修飾符
正規表示式可以包含一些可選標誌修飾符來控制匹配的模式。修飾符被指定為一個可選的標誌。我們用例項來看一下:
import re
content = '''Hello 1234567 World_This
is a Regex Demo
'''
result = re.match('^He.*?(\d+).*?Demo$', content)
print(result.group(1))
複製程式碼
和上面的例子相仿,我們在字串中加了換行符,正規表示式還是一樣的,用來匹配其中的數字。看一下執行結果:
AttributeError Traceback (most recent call last)
<ipython-input-18-c7d232b39645> in <module>()
5 '''
6 result = re.match('^He.*?(\d+).*?Demo$', content)
----> 7 print(result.group(1))
AttributeError: 'NoneType' object has no attribute 'group'
複製程式碼
執行直接報錯,也就是說正規表示式沒有匹配到這個字串,返回結果為None
,而我們又呼叫了group()
方法導致AttributeError
。
那麼,為什麼加了一個換行符,就匹配不到了呢?這是因為\.
匹配的是除換行符之外的任意字元,當遇到換行符時,.*?
就不能匹配了,所以導致匹配失敗。這裡只需加一個修飾符re.S
,即可修正這個錯誤:
result = re.match('^He.*?(\d+).*?Demo$', content, re.S)
複製程式碼
這個修飾符的作用是使.
匹配包括換行符在內的所有字元。此時執行結果如下:
1234567
複製程式碼
這個re.S
在網頁匹配中經常用到。因為HTML節點經常會有換行,加上它,就可以匹配節點與節點之間的換行了。
另外,還有一些修飾符,在必要的情況下也可以使用,如表3-3所示。
表3-3 修飾符
在網頁匹配中,較為常用的有re.S
和re.I
。
轉義匹配
我們知道正規表示式定義了許多匹配模式,如.
匹配除換行符以外的任意字元,但是如果目標字串裡面就包含.
,那該怎麼辦呢?
這裡就需要用到轉義匹配了,示例如下:
import re
content = '(百度)www.baidu.com'
result = re.match('\(百度\)www\.baidu\.com', content)
print(result)
複製程式碼
當遇到用於正則匹配模式的特殊字元時,在前面加反斜線轉義一下即可。例如.
就可以用\.
來匹配,執行結果如下:
<_sre.SRE_Match object; span=(0, 17), match='(百度)www.baidu.com'>
複製程式碼
可以看到,這裡成功匹配到了原字串。
這些是寫正規表示式常用的幾個知識點,熟練掌握它們對後面寫正規表示式匹配非常有幫助。
3. search()
前面提到過,match()
方法是從字串的開頭開始匹配的,一旦開頭不匹配,那麼整個匹配就失敗了。我們看下面的例子:
import re
content = 'Extra stings Hello 1234567 World_This is a Regex Demo Extra stings'
result = re.match('Hello.*?(\d+).*?Demo', content)
print(result)
複製程式碼
這裡的字串以Extra
開頭,但是正規表示式以Hello
開頭,整個正規表示式是字串的一部分,但是這樣匹配是失敗的。執行結果如下:
None
複製程式碼
因為match()
方法在使用時需要考慮到開頭的內容,這在做匹配時並不方便。它更適合用來檢測某個字串是否符合某個正規表示式的規則。
這裡就有另外一個方法search()
,它在匹配時會掃描整個字串,然後返回第一個成功匹配的結果。也就是說,正規表示式可以是字串的一部分,在匹配時,search()
方法會依次掃描字串,直到找到第一個符合規則的字串,然後返回匹配內容,如果搜尋完了還沒有找到,就返回None
。
我們把上面程式碼中的match()
方法修改成search()
,再看下執行結果:
<_sre.SRE_Match object; span=(13, 53), match='Hello 1234567 World_This is a Regex Demo'>
1234567
複製程式碼
這時就得到了匹配結果。
因此,為了匹配方便,我們可以儘量使用search()
方法。
下面再用幾個例項來看看search()
方法的用法。
首先,這裡有一段待匹配的HTML文字,接下來寫幾個正規表示式例項來實現相應資訊的提取:
html = '''<div id="songs-list">
<h2 class="title">經典老歌</h2>
<p class="introduction">
經典老歌列表
</p>
<ul id="list" class="list-group">
<li data-view="2">一路上有你</li>
<li data-view="7">
<a href="/2.mp3" singer="任賢齊">滄海一聲笑</a>
</li>
<li data-view="4" class="active">
<a href="/3.mp3" singer="齊秦">往事隨風</a>
</li>
<li data-view="6"><a href="/4.mp3" singer="beyond">光輝歲月</a></li>
<li data-view="5"><a href="/5.mp3" singer="陳慧琳">記事本</a></li>
<li data-view="5">
<a href="/6.mp3" singer="鄧麗君"><i class="fa fa-user"></i>但願人長久</a>
</li>
</ul>
</div>'''
複製程式碼
可以觀察到,ul
節點裡有許多li
節點,其中li
節點中有的包含a
節點,有的不包含a
節點,a
節點還有一些相應的屬性——超連結和歌手名。
首先,我們嘗試提取class
為active
的li
節點內部的超連結包含的歌手名和歌名,此時需要提取第三個li
節點下a
節點的singer
屬性和文字。
此時正規表示式可以以li
開頭,然後尋找一個標誌符active
,中間的部分可以用.*?
來匹配。接下來,要提取singer
這個屬性值,所以還需要寫入singer="(.*?)"
,這裡需要提取的部分用小括號括起來,以便用group()
方法提取出來,它的兩側邊界是雙引號。然後還需要匹配a
節點的文字,其中它的左邊界是>
,右邊界是</a>
。然後目標內容依然用(.*?)
來匹配,所以最後的正規表示式就變成了:
<li.*?active.*?singer="(.*?)">(.*?)</a>
複製程式碼
然後再呼叫search()
方法,它會搜尋整個HTML文字,找到符合正規表示式的第一個內容返回。
另外,由於程式碼有換行,所以這裡第三個引數需要傳入re.S
。整個匹配程式碼如下:
result = re.search('<li.*?active.*?singer="(.*?)">(.*?)</a>', html, re.S)
if result:
print(result.group(1), result.group(2))
複製程式碼
由於需要獲取的歌手和歌名都已經用小括號包圍,所以可以用group()
方法獲取。
執行結果如下:
齊秦 往事隨風
複製程式碼
可以看到,這正是class
為active
的li
節點內部的超連結包含的歌手名和歌名。
如果正規表示式不加active
(也就是匹配不帶class
為active
的節點內容),那會怎樣呢?我們將正規表示式中的active
去掉,程式碼改寫如下:
result = re.search('<li.*?singer="(.*?)">(.*?)</a>', html, re.S)
if result:
print(result.group(1), result.group(2))
複製程式碼
由於search()方法會返回第一個符合條件的匹配目標,這裡結果就變了:
任賢齊 滄海一聲笑
複製程式碼
把active
標籤去掉後,從字串開頭開始搜尋,此時符合條件的節點就變成了第二個li
節點,後面的就不再匹配,所以執行結果就變成第二個li
節點中的內容。
注意,在上面的兩次匹配中,search()
方法的第三個引數都加了re.S
,這使得.*?
可以匹配換行,所以含有換行的li
節點被匹配到了。如果我們將其去掉,結果會是什麼?程式碼如下:
result = re.search('<li.*?singer="(.*?)">(.*?)</a>', html)
if result:
print(result.group(1), result.group(2))
複製程式碼
執行結果如下:
beyond 光輝歲月
複製程式碼
可以看到,結果變成了第四個li
節點的內容。這是因為第二個和第三個li
節點都包含了換行符,去掉re.S
之後,.*?
已經不能匹配換行符,所以正規表示式不會匹配到第二個和第三個li
節點,而第四個li
節點中不包含換行符,所以成功匹配。
由於絕大部分的HTML文字都包含了換行符,所以儘量都需要加上re.S
修飾符,以免出現匹配不到的問題。
4. findall()
前面我們介紹了search()
方法的用法,它可以返回匹配正規表示式的第一個內容,但是如果想要獲取匹配正規表示式的所有內容,那該怎麼辦呢?這時就要藉助findall()
方法了。該方法會搜尋整個字串,然後返回匹配正規表示式的所有內容。
還是上面的HTML文字,如果想獲取所有a
節點的超連結、歌手和歌名,就可以將search()
方法換成findall()
方法。如果有返回結果的話,就是列表型別,所以需要遍歷一下來依次獲取每組內容。程式碼如下:
results = re.findall('<li.*?href="(.*?)".*?singer="(.*?)">(.*?)</a>', html, re.S)
print(results)
print(type(results))
for result in results:
print(result)
print(result[0], result[1], result[2])
複製程式碼
執行結果如下:
[('/2.mp3', '任賢齊', '滄海一聲笑'), ('/3.mp3', '齊秦', '往事隨風'), ('/4.mp3', 'beyond', '光輝歲月'), ('/5.mp3', '陳慧琳', '記事本'), ('/6.mp3', '鄧麗君', '但願人長久')]
<class 'list'>
('/2.mp3', '任賢齊', '滄海一聲笑')
/2.mp3 任賢齊 滄海一聲笑
('/3.mp3', '齊秦', '往事隨風')
/3.mp3 齊秦 往事隨風
('/4.mp3', 'beyond', '光輝歲月')
/4.mp3 beyond 光輝歲月
('/5.mp3', '陳慧琳', '記事本')
/5.mp3 陳慧琳 記事本
('/6.mp3', '鄧麗君', '但願人長久')
/6.mp3 鄧麗君 但願人長久
複製程式碼
可以看到,返回的列表中的每個元素都是元組型別,我們用對應的索引依次取出即可。
如果只是獲取第一個內容,可以用search()
方法。當需要提取多個內容時,可以用findall()
方法。
5. sub()
除了使用正規表示式提取資訊外,有時候還需要藉助它來修改文字。比如,想要把一串文字中的所有數字都去掉,如果只用字串的replace()
方法,那就太煩瑣了,這時可以藉助sub()
方法。示例如下:
import re
content = '54aK54yr5oiR54ix5L2g'
content = re.sub('\d+', '', content)
print(content)
複製程式碼
執行結果如下:
aKyroiRixLg
複製程式碼
這裡只需要給第一個引數傳入\d+
來匹配所有的數字,第二個引數為替換成的字串(如果去掉該引數的話,可以賦值為空),第三個引數是原字串。
在上面的HTML文字中,如果想獲取所有li
節點的歌名,直接用正規表示式來提取可能比較煩瑣。比如,可以寫成這樣子:
results = re.findall('<li.*?>\s*?(<a.*?>)?(\w+)(</a>)?\s*?</li>', html, re.S)
for result in results:
print(result[1])
複製程式碼
執行結果如下:
一路上有你
滄海一聲笑
往事隨風
光輝歲月
記事本
但願人長久
複製程式碼
此時藉助sub()
方法就比較簡單了。可以先用sub()
方法將a
節點去掉,只留下文字,然後再利用findall()
提取就好了:
html = re.sub('<a.*?>|</a>', '', html)
print(html)
results = re.findall('<li.*?>(.*?)</li>', html, re.S)
for result in results:
print(result.strip())
複製程式碼
執行結果如下:
<div id="songs-list">
<h2 class="title">經典老歌</h2>
<p class="introduction">
經典老歌列表
</p>
<ul id="list" class="list-group">
<li data-view="2">一路上有你</li>
<li data-view="7">
滄海一聲笑
</li>
<li data-view="4" class="active">
往事隨風
</li>
<li data-view="6">光輝歲月</li>
<li data-view="5">記事本</li>
<li data-view="5">
但願人長久
</li>
</ul>
</div>
一路上有你
滄海一聲笑
往事隨風
光輝歲月
記事本
但願人長久
複製程式碼
可以看到,a
節點經過sub()
方法處理後就沒有了,然後再通過findall()
方法直接提取即可。可以看到,在適當的時候,藉助sub()
方法可以起到事半功倍的效果。
6. compile()
前面所講的方法都是用來處理字串的方法,最後再介紹一下compile()
方法,這個方法可以將正則字串編譯成正規表示式物件,以便在後面的匹配中複用。示例程式碼如下:
import re
content1 = '2016-12-15 12:00'
content2 = '2016-12-17 12:55'
content3 = '2016-12-22 13:21'
pattern = re.compile('\d{2}:\d{2}')
result1 = re.sub(pattern, '', content1)
result2 = re.sub(pattern, '', content2)
result3 = re.sub(pattern, '', content3)
print(result1, result2, result3)
複製程式碼
例如,這裡有3個日期,我們想分別將3個日期中的時間去掉,這時可以藉助sub()
方法。該方法的第一個引數是正規表示式,但是這裡沒有必要重複寫3個同樣的正規表示式,此時可以藉助compile()
方法將正規表示式編譯成一個正規表示式物件,以便複用。
執行結果如下:
2016-12-15 2016-12-17 2016-12-22
複製程式碼
另外,compile()
還可以傳入修飾符,例如re.S
等修飾符,這樣在search()
、findall()
等方法中就不需要額外傳了。所以,compile()
方法可以說是給正規表示式做了一層封裝,以便我們更好地複用。
到此為止,正規表示式的基本用法就介紹完了,後面會通過具體的例項來講解正規表示式的用法。
本資源首發於崔慶才的個人部落格靜覓: Python3網路爬蟲開發實戰教程 | 靜覓
如想了解更多爬蟲資訊,請關注我的個人微信公眾號:進擊的Coder
weixin.qq.com/r/5zsjOyvEZ… (二維碼自動識別)