我正在尝试通过字典简化嵌套 for 循环以构建唯一值列表(嵌套字典值中列出的房间大小)。我已经将代码减少到 4 行,但很好奇是否可以通过列表推导将代码减少到 1 行。
这是一个 python 字典示例:
otas = {
Orbitz: {
u'Las Vegas': [u'1 Bedroom Suite B-side']
},
Expedia: {
u'Los Angeles': [u'2 Bedroom Lockoff', u'1 Bedroom Deluxe (A-side)', u'3 Bedroom Deluxe']
},
Priceline: {
u'New York': [u'1 Bedroom Deluxe (A-side)']
},
Travelocity: {
u'Chicago': [u'1 Bedroom Deluxe (A-side)', u'2 Bedroom Lockoff']
}
}
这是四行代码:
rooms = []
for resort in otas.values():
for room in resort.values():
rooms += [r for r in room if r not in rooms]
我知道我目前的做法没有任何错误。我很好奇是否可以做到。
最佳答案
我想您可以使用三重“嵌套”集合理解。
rooms = {roomtype
for service in otas
for location in otas[service]
for roomtype in otas[service][location]}
如果您想要返回一个列表,只需将其包装在对 list
的调用中即可。
关于python - 通过理解简化嵌套的 for 循环,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/29438530/