如何拆分熊猫数据框中的单元格并将分隔符保留在另一列中？

Question

persons
John New York
Janet New York
Mike Denver
Michelle Texas

我想分成两列：人物和城市。我试过这个：

df = pd.DataFrame({"persons": ["John New York", "Janet New York", "Mike Denver", "Michelle Texas"]})
df[["name", "city"]] = df.persons.str.split("New York", expand=True,)

它给了我这个：

          persons            name  city
0   John New York           John
1  Janet New York          Janet
2     Mike Denver     Mike Denver  None
3  Michelle Texas  Michelle Texas  None

我想要的是按城市分割并将分隔符保留在城市列中，如下所示：

          persons            name  city
0   John New York           John   New York
1  Janet New York          Janet   New York 
2     Mike Denver     Mike Denver  None
3  Michelle Texas  Michelle Texas  None

Answer 1

您可以将正则表达式与捕获组一起使用：

df[['name', 'city']] = df['persons'].str.split(r'(New York)', expand=True).iloc[:,:2]

print(df)

          persons            name      city
0   John New York           John   New York
1  Janet New York          Janet   New York
2     Mike Denver     Mike Denver      None
3  Michelle Texas  Michelle Texas      None

阅读更多关于它的工作原理here。