python - Pandas - 提取以特定字符开头的字符串

Question

它应该相当简单，但我无法实现。

我有一个数据框 df1，有一个列“name_str”。下面的例子:

   name_str 
0    alp:ha
1    bra:vo
2  charl:ie

我必须创建另一列，其中包含 - 5 个字符 - 在冒号 (:) 之后开始。我编写了以下代码:

import pandas as pd

data = {'name_str':["alp:ha", "bra:vo", "charl:ie"]}
#indx = ["name_1",]
df1 = pd.DataFrame(data=data)
n= df1['name_str'].str.find(":")+1
df1['slize'] = df1['name_str'].str.slice(n,2)
print(df1)

但输出令人失望:NaanN

   name_str  slize
0    alp:ha    NaN
1    bra:vo    NaN
2  charl:ie    NaN

输出应该是:

   name_str  slize
0    alp:ha    ha
1    bra:vo    vo
2  charl:ie    ie

有人愿意帮忙吗？欣赏它。

Answer 1

您可以使用str.extract使用以下正则表达式提取冒号后的所有内容::(.*)

df1['slize'] = df1.name_str.str.extract(':(.*)')                                                  

>>> df1                                                                                                
   name_str slize
0    alp:ha    ha
1    bra:vo    vo
2  charl:ie    ie

根据您更新的问题进行编辑

如果您想在冒号后提取最多 5 个字符，则可以使用此修改:

df['slize'] = df1.name_str.str.extract(':(.{,5})')