单行if-elif-else，具有列表推导

问题描述

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 ] 我该怎么办？

解决方法

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

,

这是dictionary的绝佳用例！

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

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

然后，使用colordict.get()检索正确的数字

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

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

使用字典会使添加更多颜色变得不容易。您能想象必须编写一百个嵌套if-elses来支持一百种颜色吗？！

使用此虚拟列表进行测试：

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

我们得到了输出

[1,3,2,3]

这正是我们所期望的。

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