在给定列中拆分字符串值（熊猫）

Posted 2023-03-12

【中文标题】在给定列中拆分字符串值（熊猫）

【英文标题】：Splitting string value in given column (Pandas)

【发布时间】：2021-08-05 16:05:52

【问题描述】：

我有一个包含许多这样的行的数据框：

ID	Variable
1	A1_1 - Red
2	A1_2 - Blue
3	A1_3 - Yellow

我正在尝试遍历所有行，以便所有第二列的值都更改为“A1”。我想出的代码是：

for row in df.iterrows():

  current_response_id=row[1][0]

  columncount=0

  for columncount in range(2):

    variable=row[1][1];

    row[1][1]=variable.split("_")[0].split(" -")[0]

    variable=row[1][1];

但是，这并没有达到预期的效果。我该怎么办？

【参考方案1】：

试试：

df["Variable"] = df["Variable"].str.split("_").str[0]
print(df)

打印：

   ID Variable
0   1       A1
1   2       A1
2   3       A1

【讨论】：

效果很好！谢谢！但是，我仍然不明白为什么我的初始代码没有。