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

例子

基本用法

考慮以下 DataFrame ：

import pandas as pd
df = pd.DataFrame({'A':['k23','45k','67k89']}, index=['a','b','c'])
df



   A
a  k23
b  45k
c  67k89

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

df['A'].str.extractall('(\d+)')   # returns a multi-index DataFrame



          0
match  
a      0  23
b      0  45
c      0  67
       1  89

這裏，輸入字符串是一個正則表達式，\d+表示一個數字，而()表示我們要提取的部分。

由於生成的 DataFrame 是一個多索引，因此我們可以獲得特定索引的匹配，如下所示：

df_result = df['A'].str.extractall('(\d+)')
df_result.loc['c']



        0
match  
0       67
1       89

多個捕獲組

考慮以下 DataFrame ：

import pandas as pd
df = pd.DataFrame({'A':['k23','45y','67k89']}, index=['a','b','c'])
df



   A
a  k23
b  45y
c  67k89

我們可以使用多個括號捕獲多個組：

df['A'].str.extractall('(\d+)([ky])')



            0   1
    match    
b     0     45  y
c     0     67  k