python - 在 python 中的下划线处拆分并存储第一个值

Question

我有一个像 df 这样的 pandas 数据框，其中有一列 construct_name

construct_name
aaaa_t1_2    
cccc_t4_10
bbbb_g3_3

等等。我想先将下划线处的所有名称拆分，并将第一个元素(aaaa，cccc等)存储为另一个列名。

预期输出

construct_name  name
aaaa_t1_2       aaaa
cccc_t4_10      bbbb

等等。

我尝试了以下df['construct_name'].map(lambda row:row.split("_")) 它给了我一个像这样的列表

[aaaa,t1,2]
[cccc,t4,10]

等等

但是当我这样做的时候

df['construct_name'].map(lambda row:row.split("_"))[0] 获取列表的第一个元素时出现错误。你能建议修复吗？谢谢

Answer 1

只需使用向量化的 str方法 split并在列表上使用整数索引来获取第一个元素:

In [228]:

df['first'] = df['construct_name'].str.split('_').str[0]
df
Out[228]:
  construct_name first
0      aaaa_t1_2  aaaa
1     cccc_t4_10  cccc
2      bbbb_g3_3  bbbb