python - Python Pandas 中的数据提取

Question

我正在使用 Pandas 的 groupby 函数从 excel 数据中获取结果。我的代码是:

results = df.groupby(['Test Result', 'Risk Rating']).size().unstack(fill_value=0)

它以表格的形式给我结果:

Risk Rating  CRITICAL  HIGH  LOW  MEDIUM
Test Result
FAIL                8     0    9       4
PASS                0    13   23      37
SKIP                2     0    0       0

现在，我只想得到应该给我的所有“失败”的结果:

Risk Rating  CRITICAL  HIGH  LOW  MEDIUM
Test Result
FAIL                8     0    9       4

而且，只有 CRITICAL 的“FAIL”应该给我:

Risk Rating  CRITICAL
Test Result
FAIL                8    

我怎样才能做到这一点。任何帮助将不胜感激。

Answer 1

使用.loc

results.loc['Fail']

将返回索引“失败”的值

results.loc['Fail', 'Critical']

将为“关键”列返回索引“失败”的值

您可以为多列或索引传递列表，例如:

results.loc['Fail', ['Critical','HIGH']]