带有列表推导的单行 if-elif-else答案

【问题标题】：Single-line if-elif-else with list comprehensions带有列表推导的单行 if-elif-else
【发布时间】：2021-01-29 12:44:36
【问题描述】：

item 是 MySQL 中的所有元组

y=[1 if (t['color']=='yellow') else -1 for t in item]

输出：

[-1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1  -1 -1 -1 -1 -1 -1  1  1  1  1  1  1  1  1  1  1  1  1  1  1  1 ]

喜欢这个，我想做这个

for t in item:
   if(t['color']=='yellow'):
       y=1
   elif(t['color']=='blue'):
       y=2
   else:
       y=3

并将结果放入数组中

y=[1 if (t['color']=='yellow') 2 elif (t['color']=='blue') else 3 for t in item]
y=np.array(y)
print(y)

输出应该是这样的[ 1 2 3 1 3 2 1 3 1 2 3 2 1 ] 我该怎么做？

【问题讨论】：

标签： python python-3.x list-comprehension conditional-operator

【解决方案1】：

这是dictionary 的一个很好的用例！

首先，定义一个将颜色名称映射到数字的字典

colordict = {'yellow': 1, 'blue': 2, 'default': 3}

然后，使用colordict.get() 检索正确的号码

y = [colordict.get(t['color'], colordict['default']) for t in item]

.get() 的第二个参数是在colordict 中找不到t['color'] 时返回的默认值。

使用 dict 可以轻松添加更多颜色。你能想象必须写一百个嵌套的if-elses 来支持一百种颜色吗？！

使用这个虚拟列表进行测试：

item = [
    {'color': 'yellow', 'data': 0.5},
    {'color': 'purple', 'data': 0.1}, 
    {'color': 'blue', 'data': 0.2}, 
    {'color': 'blue', 'data': 0.3}, 
    {'color': 'red', 'data': 0.6}
       ]

我们得到输出

[1, 3, 2, 2, 3]

这正是我们所期望的。

如果您不想处理.get()，也可以使用defaultdict。

【讨论】：

【解决方案2】：

y=[1 if (t['color']=='yellow') else 2 if (t['color']=='blue') else 3 for t in item]

【讨论】：