填充列表列表以使其等于最大列表的大小

Question

我有一个句子列表列表，我想填充所有句子，使它们的长度相同。我能够做到这一点，但我正在努力寻找最佳的做事方式和挑战自己。max_length = max(len(sent) for sent in sents)list_length = len(sents)sents_padded = [[pad_token for i in range(max_length)] for j in range(list_length)]for i,sent in enumerate(sents):    sents_padded[i][0:len(sent)] = sent 我使用了输入：sents = [["Hello","World"],["Where","are","you"],["I","am","doing","fine"]]pad_token = "Hi"我的方法是一种有效的方法还是有更好的方法？

Answer 1

这是在 itertools（在 python3 中）中提供的，用于迭代，带有 zip_longest，您可以使用 zip(*) 正常反转它，如果您更喜欢它而不是迭代器，则将其传递给列表。

import itertools

from pprint import pprint

sents = [["Hello","World"],["Where","are","you"],["I","am","doing","fine"]]

pad_token = "Hi"

padded = zip(*itertools.zip_longest(*sents, fillvalue=pad_token))

pprint (list(padded))

[['Hello', 'World', 'Hi', 'Hi'],

['Where', 'are', 'you', 'Hi'],

['I', 'am', 'doing', '美好的']]

Answer 2

以下是如何使用str.ljust()填充每个字符串，并使用max()keylen来查找填充每个字符串的数字：

lst = ['Hello World', 'Good day!', 'How are you?']

l = len(max(lst, key=len)) # The length of the longest sentence

lst = [s.ljust(l) for s in lst] # Pad each sentence with l

print(lst)

输出：

['Hello World ',

 'Good day!   ',

 'How are you?']

Answer 3

假设：

输出应与 OP 输出相同（即每个子列表中的单词数相同）。

输入：

sents = [["Hello","World"],["Where","are","you"],["I","am","doing","fine"]]

pad_token = "Hi"

以下 1-liner 产生与 OP 代码相同的输出。

sents_padded = [sent + [pad_token]*(max_length - len(sent)) for sent in sents]

print(sents_padded)

# [['Hello', 'World', 'Hi', 'Hi'], ['Where', 'are', 'you', 'Hi'], ['I', 'am', 'doing', 'fine']

Answer 4

当我计时时，这似乎更快：

maxi = 0

for sent in sents:

    if sent.__len__() > maxi:

        maxi = sent.__len__()

for sent in sents:

    while sent.__len__() < maxi:

        sent.append(pad_token)

print(sents)