在遍历使用pandas数据框的列表时,可以使用apply()
函数来进行字符串查找。下面是一个示例代码:
import pandas as pd
# 创建一个包含字符串的数据框
df = pd.DataFrame({'column1': ['abc', 'def', 'ghi'],
'column2': ['xyz', 'lmn', 'opq']})
# 定义一个函数,用于查找字符串
def find_string(text, keyword):
if keyword in text:
return True
else:
return False
# 使用apply()函数遍历数据框的每一行,查找特定的字符串
keyword = 'de'
df['found'] = df.apply(lambda row: find_string(row['column1'], keyword), axis=1)
print(df)
输出结果为:
column1 column2 found
0 abc xyz False
1 def lmn True
2 ghi opq False
在这个例子中,我们定义了一个find_string()
函数,用于在给定的字符串中查找特定的关键字。然后,我们使用apply()
函数遍历数据框的每一行,并调用find_string()
函数来查找关键字。最后,我们将查找结果存储在新的列"found"中。