python - 如何在 pandas loc 过滤条件中传递 "Take All"参数？

Question

我有一个带有参数(在本例中为“部门”)的函数，用于从我的数据集中过滤 (df.loc[(df['A'] == department) 特定数据。在一种情况下，我想要要使用此特定功能，但我不想过滤数据，而是想获取所有数据。

有没有办法传递一个参数，这会导致类似df.loc[(df['A'] == *) 或df.loc[(df['A'] == %)

    # Write the data to the table 

    def table_creation(table, department, status):

        def condition_to_value(df, kpi):
            performance_indicator = df.loc[(df['A'] == department) & (df['C'] == kpi) & (df['B'] == status), 'D'].values[0]
        return performance_indicator

Answer 1

我能想到的一种方法是，您可以使用 df['A'].isin(['部门'])。两者产生相同的结果。

一旦你这样做了，你就可以像这样传递“Take All”参数:

df['A'].isin(df['A'].unique())

其中 df['A'].unique() 是此列中所有唯一参数的列表，因此它将返回所有 True。

或者你可以像这样传递多个参数:

df['A'].isin(['department', 'string_2', 'string_3']))