在散点图python matplotlib中创建颜色图例

Question

我正在尝试使用以下代码：

color= ['red' if House =="20" "blue" elif House =="21" "yellow" elif House =="22" else 'green' for H in House]

它给出以下错误： SyntaxError：语法无效

任何建议 - 想法？

Answer 1

elif不能用于列表理解。它应该是else <value> if <condition>。应用于您的代码：

color= ['red' if H=="20" else "blue" if H =="21" else "yellow" if H =="22" else 'green' for H in House]

请注意，您正在迭代H，而不是House

Answer 2

如果你使用字典会更容易阅读：

Projet