文章目录
在DataFrame中某列插入数据,并根据另一列查找是否存在某个字符串完全一样,在另一列插入对应数据的功能
可以使用pandas的loc
方法来实现在DataFrame中某列插入数据,并根据另一列查找是否存在某个字符串完全一样,在另一列插入对应数据的功能。假设要在DataFrame中的column1
列中插入数据,并根据column2
列查找是否存在某个字符串完全一样,在column3
列插入对应数据,代码如下:
import pandas as pd
# 创建一个示例DataFrame
df = pd.DataFrame({'column1': ['A', 'B', 'C', 'D'], 'column2': ['abc', 'def', 'abc', 'ghi']})
# 要插入的数据
new_data = ['X', 'Y', 'Z']
# 要查找的字符串
search_str = 'abc'
# 在column1列中插入数据,并根据column2列查找是否存在某个字符串完全一样,在column3列插入对应数据