“我有一个数据框,其中包含活动名称,地址,经度,纬度,日期,......等列 “我想创建一个函数,使我能够使用条件过滤数据集。
如果“活动名称”列中存在某些文本。我想查看过滤后的数据集并将其分配给新的数据集”
dfnew = []
def update_text(selected_text):
for text in df['Activity Name']:
if selected_text in text:
dfnew == df[df["Activity Name"].str.contains(selected_text)]
答案 0 :(得分:0)
尝试一下:
def update_text(df, selected_text):
dfnew = []
for row in df:
if selected_text in row[col_idx]:
dfnew.append(row)
return dfnew
dfnew = update_text(df, "some text")
print(dfnew)
该函数遍历各行,如果“活动名称”列中的值包含selected_text,则将该行追加到dfnew。 col_idx应该是“活动名称”列的编号索引。