根据现有列表创建一个新列表

Question

我有几个与每个颜色相对应的列表，我想生成颜色类型。

Colour = ['Red, Yellow', 'Orange, Black', 'Blue, Green', 'Red']

我想生成一个与BallColour相同的列表，但具有颜色类型（如下所示）：

ColourType = ['Primary, Primary', 'Secondary, Unknown', 'Primary, Secondary', 'Primary'] 

Answer 1

使用dict轻松做到这一点：

BallColour = ['Red, Yellow', 'Orange, Black', 'Blue, Green', 'Red']

c_key = dict(
    Red = 'Primary',
    Blue = 'Primary',
    Yellow = 'Primary',
    Green = 'Secondary',
    Orange = 'Secondary',
    Purple = 'Secondary',
)

ColourType = []
for colors in BallColour:
    colors = colors.split(', ')
    temp = []
    for color in colors:
        if color in c_key:
            temp.append(c_key[color])
        else:
            temp.append('Unknown')
    ColourType.append(', '.join(temp))

print(ColourType)

example in python tutor