创建一个函数，用于从数据集列中的字符串中获取子串
提问于 2019-05-22 19:49:04
我有一个数据集，其中一列是文本。我想创建一个函数来获取基于输入的字符串(如果该字符串包含该子字符串，则打印该字符串)。我已经写了一个函数，但是不确定为什么它不能工作。
def update_text("selected_text"):
for text in df['Activity Name']:
if "selected_text" in text:
print(text)
回答 1
Stack Overflow用户
回答已采纳
发布于 2019-05-22 20:02:16
我用样本数据进行了测试，如下所示：
import pandas as pd
def update_text(selected_text):
for text in df['Activity Name']:
if selected_text in text:
print(text)
df = pd.DataFrame(['hello monkey', 'welcome'], columns=['Activity Name'])
update_text('hello')结果如下：
hello monkey对于新问题：
import pandas as pd
def update_text(df, selected_text):
dfnew = df.loc[df['Activity Name'].str.contains(selected_text)]
return dfnew
df = pd.DataFrame(['hello monkey', 'welcome'], columns=['Activity Name'])
dfnew = update_text(df, 'hello')
print (dfnew) Activity Name
0 hello monkey页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/56263988
复制
腾讯云开发者
