时间戳

首页 > 解决方案 > 根据另一列 pandas 中字符串的特定字符创建新的数据框列

python - 根据另一列 pandas 中字符串的特定字符创建新的数据框列

问题描述

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

   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

有可能做到吗？

提前致谢

标签： pythonpandasdataframe

解决方案

使用str.endswith()

前任：

df = pd.DataFrame({"Col1": ['016e3d588c', '071b4x718g', '011e3d598c', '041e0i608g']})
df["Col2"] = df["Col1"].str.endswith("g").astype(int)
print(df)

输出：

         Col1  Col2
0  016e3d588c     0
1  071b4x718g     1
2  011e3d598c     0
3  041e0i608g     1

推荐阅读