列表理解

Question

我有一个for循环用于将值输出到列表中。

这是我的输出：

[[23, 34, 34] [34,21,34,56] [21,3,5,67]]

以下是适用于上述输出的代码：

y_train = ([[word2index[w] for w in sent[1:]] for sent in tokenized_sentences]).

但我想在每个较小的列表的末尾添加一个值。如何修改我的代码来处理这个？我想要的输出应该是这样的：

[[23,34,34,**2**][34,21,34,56,**2**][21,3,5,67,**2**]]

所以我想在每个内部列表的末尾附加一个新值。

P.S。处理此问题的正常for循环也会很好。

Answer 1

for x in y_train:
    x.append(element)

示例：

>>> listOfLists = [[1,2], [2,3], [4,5]]
>>> for x in listOfLists:
...     x.append(2)
>>> listOfLists
[[1, 2, 2], [2, 3, 2], [4, 5, 2]]