Как использовать значения idx, возвращаемые nLargest () в Pandas во втором фрейме данных?

Question

Я хотел бы найти n самых больших чисел в одном фрейме данных и взять соответствующие значения во втором фрейме данных.

Код ниже показывает, что я пытаюсь сделать.Возможно ли это сделать?

df = pd.DataFrame({'a': [1, 10, 8, 20, -1]})
df2 = pd.DataFrame({'a': [3, 20, 4, 11, 14]})

df.nlargest(2, 'a')
df2[df2 == df.nlargest(2, 'a')]

Answer 1

Попробуйте:

df2.loc[df.nlargest(2, 'a').index]

Это будет работать, если ваши кадры данных проиндексированы одинаково.

Вывод:

    a
3  11
1  20