首页 > 解决方案 > python更新csv文件,其中column2以column1开头

问题描述

我想在新的 column3 中打印 true,其中 column2 从 column1 开始,否则在 column3 中打印 false

这是我的示例 CSV 文件

column1  column2
foo      footer
one      two
pro      programmer
queue    text

我想要更新的 csv 文件,如下所示

column1  column2     column3
foo      footer      true
one      two         false
pro      programmer  true
queue    text        false

谢谢

标签: pythonpandas

解决方案


使用 alist comprehension检查 column1 是否存在于第 2 列中

df['column3'] = [column2.startswith(column1) for column1, column2 in df.to_numpy()]

推荐阅读