python - Pandas - 用不同的长度对每一行进行子串

Question

你好，

我有一个数据框，我想在其中为该列的每一行隔离一部分字符串。我遇到的问题是每一行都需要有一个不同长度的子字符串，特别是我只想保留字符串直到第一次出现“。”。 (句点)加上接下来的两个字母。

例子:

import pandas as pd

x = [ [ 34, 'Sydney.Au123XX'] ,
             [30, 'Delhi.As1q' ] ,
             [16, 'New York.US3qqa']]
x = pd.DataFrame(x)
x.columns = ["a", "b"]

#now I want to substring each row based on where "." occurs.
#I have tried the following:
y = x["b"].str.slice( stop = x["b"].str.find(".") + 2)
y = x["b"].str[0: x["b"].str.find(".")+ 2]

#desired output
desired = [[ 34, 'Sydney.Au'] ,
             [30, 'Delhi.As' ] ,
             [16, 'New York.US'] ]
desired  = pd.DataFrame(desired )
desired .columns = ["a", "b"] 

请查看我的代码以获得所需的输出。

我不想使用循环。

提前致谢。

Answer 1

IIUC 尝试:

x['b'] = x['b'].str.split('.').str[0]
print(x)

你也可以做一个单行:

print(x.assign(b=x['b'].str.split('.').str[0]))

它们都输出:

    a         b
0  34    Sydney
1  30     Delhi
2  16  New York

编辑:

做:

x['b'] = x['b'].str.extract('(.*\...)')
print(x)

或者使用:

print(x.assign(b=x['b'].str.extract('(.*\...)')))