Select Python 列表列表中的前 n 个元素

Question

我正在尝试根据现有列表创建一个新列表。我正在尝试 select 内部列表中的前 N 个值并输出新的列表列表。当前列表如下所示：

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

现在我想将其作为输入并输出每个列表中的前 1 个元素。真正的列表更长，有 2 个以上的列表，我想取前 N 个元素。输出应如下所示：

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

如有任何帮助，我们将不胜感激！

Answer 1

使用带有内部列表切片的列表理解。

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

Select Python 列表列表中的前 n 个元素

Select top n elements from list of lists in Python

python

loops

list

nested-lists

python-3.x