从一列复制字符串的一部分并将其放入新的 pandas 列

Question

我正在清理数据集，需要取下划线 (_) 之间的字符串部分。 A列是我开始的。

A   
foo_bar_foo
bar_foo_bar
bar
foo_bar_foo

我需要复制下划线之间的字符并将它们复制到一个新列中。 B 列是预期结果。

A               B
foo_bar_foo     bar
bar_foo_bar     foo
bar             null
foo_bar_foo     bar

任何建议表示赞赏。

Answer 1

使用str.split和.str[index]

df['B']=df.A.str.split('_').str[1]



      A        B
0  foo_bar_foo  bar
1  bar_foo_bar  foo
2          bar  NaN
3  foo_bar_foo  bar

Answer 2

使用提取物：

df['B'] = df['A'].str.extract('_(\w+)_')
print(df)

Output

             A    B
0  foo_bar_foo  bar
1  bar_foo_bar  foo
2          bar  NaN
3  foo_bar_foo  bar