python - 如何从 pandas 的每一列中获取 n 个最多的列值

Question

我知道如何使用“模式”获取数据框中每一列的最频繁值。例如:

df = pd.DataFrame({'A': [1, 2, 1, 2, 2, 3]})
df.mode()
   A
0  2

但我无法找到数据框每一列的“n”个最常见值？例如对于提到的数据框，我想要以下 n=2 的输出:

   A
0  2
1  1

任何指针？

Answer 1

一种方法是使用 pd.Series.value_counts 并提取索引:

df = pd.DataFrame({'A': [1, 2, 1, 2, 2, 3]})

res = pd.DataFrame({col: df[col].value_counts().head(2).index for col in df})

#    A
# 0  2
# 1  1