从 Python 中的列表列表中选择前 n 个元素

Question

我正在尝试从现有列表创建一个新列表。我试图从内部列表中选择前 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]