python - pandas groupby 具有特定值的列

Question

嗨，我正在使用 pandas 中的数据集。假设数据集有ID，测试类型，测试状态，测试日期，等等

我需要对一种列进行分组，所以我首先尝试

data_useless[['TEST_TYPE', 'TEST_STATUS']].groupby('TEST_STATUS').count_values()

它起作用了:显示按 test_status 分组数据的结果(失败的测试和通过测试)以及数据集上该数据的计数值

现在我想了解并查看通过测试的更多数据

所以我尝试了

data_useless.groupby(['TEST_STATUS'] == 'PASS TEST')

它不起作用..显示错误，它说KEY ERROR:FALSE

我需要做类似的事情在 SQL 中:

SELECT *
FROM data_useless
WHERE TEST_STATUS = "PASS TEST"
group by TEST_STATUS;

请帮忙,,,

Answer 1

您可以使用以下方式进行过滤:

data_useless.loc[data_useless['TEST_STATUS'] == 'PASS TEST']

或者:

data_useless.query('TEST_STATUS == "PASS TEST"')

然后，如果需要，计算 groupby + 聚合:

(data_useless.loc[data_useless['TEST_STATUS'] == 'PASS TEST']
             .groupby(...).agg(...) # or any other groupby function
)