如何遍历满足某些条件的 Pandas 数据框并将这些行附加到新数据框？

Question

我正在尝试创建一个函数 (get_filtered_df)，它给定一个过滤器和一个数据集，遍历特定列并应用一些函数 (get_filter) 来查找满足这些条件的行并添加到某个最终数据框中。

这是我到目前为止所完成的：

def get_filter(filter):
    string_container = []
    
    string_container.append(filter.upper())
    string_container.append(filter.lower())
    # More conditions here
    
    return (string_container)

def get_filtered_df(string_container, df):
    filtered_df = pd.DataFrame(df[df['Titles'].str.contains(string_container[0])], columns=df.columns)
    
    return(filtered_df)

当我尝试使用 for 或 while 循环而不是固定索引 (0) 进行迭代时，问题就出现了，因为我从来没有得到函数来附加满足条件的行。

这是我尝试使用的具体说明：

for i in range len(get_filter('star')):
    filtered_df.append(df[df['Title'].str.contains(get_filter('star')[i])])

Filtered_df 返回始终为空

Answer 1

pd.DataFrame.append方法返回一个新对象。 在您的代码中，您要附加一个新的 Dataframe - 它在每次迭代时都会丢失，因此返回一个空的 DataFrame。

您需要更新filtered_df在你的循环，以更新filtered_df ：

def get_filtered_df(string_container, df):
    filtered_df = pd.DataFrame(data=None, columns=df.columns)
    for i in range(len(string_container)):
        filtered_df = filtered_df.append(df[df['Titles'].str.contains(string_container[i])], ignore_index=True)
    return(filtered_df)

应该管用。