Python Pandas：拆分列并在当前[重复]旁边添加新列

Question

我有一个类似于此的excel工作表，但有很多列：

我想将团队列拆分为团队和一个名为Team ID的新列。我目前使用以下代码执行此操作：

df[['Team', 'Team ID']] = df['Team'].str.split(r"\s\(+(?=\S*$)", expand=True)
df['Team ID'] = df['Team ID'].str[:-1]

这很好（请注意，团队名称可以包括数字、空格和paranthesis）。虽然这可能并不完美，但我完成了这项工作。

我的问题是新列“Team ID”放置在数据集的末尾。所以它将是“Team-成员-Team ID”。虽然3列不是问题，但有时有10列需要拆分7列。

所以问题是：有没有办法把一列一分为二，然后把新创建的列放在旧列的旁边？

Answer 1

您可以将str.extract与正则表达式一起使用。

要插入到正确的位置，可以使用插入：

out = df['Team'].str.extract('(\w+) \((\d+)\)')

df['Team'] = out[0]
df.insert(df.columns.get_loc('Team')+1, 'Team ID', out[1])

输出：

    Team  Team ID  Members
0  Team1      553    95435
1  Team2      443      872

正则表达式：

(\w+)      # match word
\((\d+)\)  # match digits surrounded by parentheses