从 Python 中的字符串中提取第二个字符

Question

我有一个 dataframe 列，其中包含某些值，我想在连字符之后提取第二个字符串并保留原始值。

数据

   type            stat
   SSS-AA-11111    y
   FFF-BB-22222    y

期望的

type            type1 stat
SSS-AA-11111    AA    y
FFF-BB-22222    BB    y

正在做

  df[['type', 'type1']] = df['type'].str.split('-', 1, expand=True)`

我收到此错误：

ValueError: Columns must be same length as key

我还在研究。 任何建议表示赞赏

Answer 1

使用str.split并使用str[1]保留第二个元素：

df['type1'] = df['type'].str.split('-').str[1]
print(df)

# Output
           type stat type1
0  SSS-AA-11111    y    AA
1  FFF-BB-22222    y    BB