如何将列表数组转换为字符串集合，并将字符串乘以数字（提供的示例）

Question

我得到了以下列表数组

[array(['I', '1', 'am', '1', 'a', '1', 'boy', '3']) array(['I', '1', 'am', 'a', 'girl', '1']) array(['I', '1', 'am', '1', 'non-binary', '2'])

我想编写一个python代码，将数组转换为语料库，其中文本乘以数字，即以下格式：

corpus = [ "I am a boy boy boy", "I am a girl", "I am non-binary non-binary]

如何在Python上实现这一点？任何帮助都将不胜感激

Answer 1

您可以使用列表理解来获取所有单词，然后使用另一个列表理解来“展平”列表

data = [['I', '1', 'am', '1', 'a', '1', 'boy', '3'], 
        ['I', '1', 'am', '1', 'a', '1', 'girl', '1'], 
        ['I', '1', 'am', '1', 'non-binary', '2']]

corpus = [[word]*int(sentence[pos+1]) for sentence in data for pos, word in enumerate(sentence) if pos % 2 == 0]

corpus = [word for words in corpus for word in words]

print(corpus)

""" OUTPUT
['I', 'am', 'a', 'boy', 'boy', 'boy', 'I', 'am', 'a', 'girl', 'I', 'am', 'non-binary', 'non-binary']
"""