python - 获取 pandas 数据对象的 x,y 样式坐标

Question

我想知道是否可以在 pandas 数据框中搜索值，如果找到，则返回行、列名称样式坐标。例如，如果在下面的数据帧中搜索 mat 字符串，则将返回 (3, C)。

     A    B    C
0   20     32  bat
1   30    234  cat
2   10     23  rat
3   40     23  mat
4   50  42523  etc

我已经通读了文档，我发现的最接近的解决方案是 index.get_loc() ，尽管这只返回基于命名索引的 loc 。

Pandas 版本0.22.0

Answer 1

正如 @Wen 所建议的，np.where 是搜索值坐标的有效方法。下面的方法将返回此类坐标的列表。

import pandas as pd

df = pd.DataFrame([[20, 32, 'bat'], [30, 234, 'cat'],
                  [10, 23, 'rat'], [40, 23, 'mat'],
                  [50, 42523, 'etc']], columns=['A', 'B', 'C'])

[(x, df.columns[y]) for x, y in zip(*np.where(df.values == 'mat'))]  # [(3, 'C')]