基于另一列中字符串的特定字符创建新的dataframe列

Question

我有一个包含字符串的列的数据帧。我想知道是否可以基于dataframe创建一个新列。 这是列的一个示例：

   col1
016e3d588c
071b4x718g
011e3d598c
041e0i608g

我想根据字符串的最后一个字符创建一个新列。这就是我所尝试的：

for i in DF['col1']:
    if i[-1] == 'g':
        DF['col2'] = 1
    else:
        DF['col2'] = 0

我想要这样的新专栏：

col2
  0
  1
  0
  1

但我的代码有以下输出：

col2
  0
  0
  0
  0

有可能吗

提前谢谢

Answer 1

您可以使用numpy尝试此操作

import numpy as np

DF["col2"] = np.where(DF["col1"].str[-1]=="g",1,0)