Выберите верхнюю 2 строку каждого элемента в мультииндексном фрейме данных в pandas - PullRequest
0 голосов
/ 06 марта 2020

У меня есть мультииндексный фрейм данных, и я хочу выбрать 2 верхние строки из каждого элемента на уровне 1. Это не фрейм данных, а как пример >> скажем, этот фрейм данных

mux = pd.MultiIndex.from_arrays([
list('aaabbbcccddd'),
list('tuvwacdjpkqz')
], names=['level 1', 'level 2'])
df = pd.DataFrame({'col': np.arange(len(mux))}, mux)

I хотел бы получить следующий результат, полученный при запуске этого фрейма данных. Итак, первые 2 элемента из каждой записи на уровне 1. Это всего лишь пример выходных данных. Я не хочу заново создавать фрейм данных, а выбираю первые две строки из первой

emux2 = pd.MultiIndex.from_arrays([
list('aabbccdd'),
list('tuwadjkq')], names=['level 1', 'level 2'])
df2 = pd.DataFrame({'col': np.arange(len(mux2))}, mux2)

Как я могу нарезать фрейм данных df, чтобы получить эти результаты в df2?

...