根据数据框中某一列的文本选择条件,从现有数据框中创建一个新数据框

时间:2019-05-23 23:15:47

标签: python

“我有一个数据框,其中包含活动名称,地址,经度,纬度,日期,......等列 “我想创建一个函数,使我能够使用条件过滤数据集。

如果“活动名称”列中存在某些文本。我想查看过滤后的数据集并将其分配给新的数据集”

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)]

1 个答案:

答案 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应该是“活动名称”列的编号索引。