Python Pandas Series str extract方法用法及代碼示例

例子

基本用法

考慮以下 DataFrame ：

import pandas as pd
df = pd.DataFrame({'A':['a1','b2','c3']})
df



   A
0  a1
1  b2
2  c3

要獲取與給定正則表達式匹配的提取子字符串：

df['A'].str.extract('[ab](\d+)')



   0
0  1
1  2
2  NaN

這裏，[ab]表示a或b，\d+表示數字。我們使用()來指示我們要提取的部分。

多個捕獲組

我們可以使用多個括號捕獲多個組，如下所示：

df['A'].str.extract('([ab])(\d+)')   # returns a DataFrame



   0    1
0  a    1
1  b    2
2  NaN  NaN

設置展開

考慮以下 DataFrame ：

import pandas as pd
df = pd.DataFrame({'A':['a1','b2','c3']})
df



   A
0  a1
1  b2
2  c3

默認情況下， expand=True ，這意味著即使隻有一個捕獲組，也會返回 DataFrame ：

df['A'].str.extract('[ab](\d+)')   # expand=True



   0
0  1
1  2
2  NaN

要獲取係列(或索引)，請設置 expand=False ：

df['A'].str.extract('[ab](\d+)', expand=False)



0      1
1      2
2    NaN
Name: A, dtype: object