从Pandas数据框中提取符合标准的细胞指标

Question

我在熊猫身上有这样一个数据框：

  a b c
A 1 2 3 
B 4 5 6
C 7 8 9

我想提取其单元格的值大于6的索引名和列名对。在

换句话说，我想得到

^{pr2}$

有什么聪明的方法吗？在

Answer 1

您可以使用^{}将框架展平为一个系列，使用布尔索引选择所需的术语，最后将生成的索引转换为列表：

s = df.stack()
ii = s[s >= 6].index.tolist()

例如：

^{pr2}$

请注意（1）我使用>= 6，因为这与您的示例相匹配，（2）这是严格意义上的元组列表，而不是您所要求的列表，但是如果您真的想转换，可以转换。在

Answer 2

您可以选择NumPy路线并使用^{}和{a2}。在

生成行和列标签的数组，并使用np.argwhere返回给定条件所在的整数索引。使用np.choose检索相应的行和列索引标签：

>>> xy = np.column_stack((df.index, df.columns))
>>> np.choose(np.argwhere(df >= 6), xy)
array([['B', 'c'],
       ['C', 'a'],
       ['C', 'b'],
       ['C', 'c']], dtype=object)

这将返回一个数组：您可以使用tolist()来获取Python列表：

^{pr2}$