Выберите лучшие n элементов из списка списков в Python

Question

Я пытаюсь создать новый список списков из существующего. Я пытаюсь выбрать N верхних значений из внутренних списков и вывести новый список списков. Текущий список выглядит следующим образом:

[[('with', -3.608809242675524),
  ('cruz', -3.6553518286125546),
  ('rt', -2.962847001582534)],
[ ('his', -2.9577717646107025),
  ('is', -2.9627189839205403),
  ('to', -2.968666793195209)]]

Теперь я хочу взять это как входные данные и вывести первый элемент из каждого списка. Реальный список длиннее и имеет более 2 списков, и я хотел бы взять верхние N элементов. Вывод должен выглядеть следующим образом:

[[('with', -3.608809242675524)],
[('his', -2.9577717646107025)]]

Любая помощь будет оценена!

Answer 1

Использовать понимание списка с нарезкой внутренних списков.

result = [l[:N] for l in original_list]