Python以数组增量将值追加到列表

Question

假设我有一个列表列表：

name_list = ['John', 'Smith'], ['Sarah', 'Nichols'], ['Alex', 'Johnson']

我还有另一个任意列表，让我们说一下颜色：

color_list = ['Red', 'Green', 'Blue']

将指定数量的颜色附加到旋转名称中最简单或最有效的方法是什么？ 例如，如果我选择两种颜色，则返回的列表将如下所示：

output_list = ['John', 'Smith', 'Red'], ['Sarah', 'Nichols', 'Green'], ['Alex', 'Johnson', 'Red']

或者如果我选择3种颜色：

output_list = ['John', 'Smith', 'Red'], ['Sarah', 'Nichols', 'Green'], ['Alex', 'Johnson', 'Blue']

Answer 1

您可以itertools.cycle列表的一部分：

name_list = [['John', 'Smith'], ['Sarah', 'Nichols'], ['Alex', 'Johnson']]


color_list = ['Red', 'Green', 'Blue']

from itertools import cycle, islice

i = 2 
cyc =  cycle(islice(color_list, 0, i))
for  sub in name_list:
    sub.append(next(cyc))


print(name_list)

输出：

[['John', 'Smith', 'Red'], ['Sarah', 'Nichols', 'Green'], ['Alex', 'Johnson', 'Red']]

你也可以使用索引，我们遍历name_list模i来提取正确的元素color_list ：

name_list = [['John', 'Smith'], ['Sarah', 'Nichols'], ['Alex', 'Johnson'], ['Alex', 'Johnson']]

color_list = ['Red', 'Green', 'Blue']


i = 2
for ind, sub in enumerate(name_list):
    sub.append(color_list[ind % i])

print(name_list)

Answer 2

首先-您的列表列表应如下所示：

name_list = [['John', 'Smith'], ['Sarah', 'Nichols'], ['Alex', 'Johnson']]

您缺少括号...

一种简单的方法是使用模运算符...（检查边缘情况，例如当颜色列表为空时）：

color_len = len(color_list)
for idx,elem in enumerate(name_list):
    elem.append( color_list[idx%color_len] )