如何在python中对生成器类型进行排序

Question

我的代码有以下几行

get_alarm_list = conn.query_alarms(query.filter_expr,
                                   query.orderb) 
print "type is:", type(get_alarm_list)
for alarm in get_alarm_list:
    if alarm.severity == 'critical':
        alarm.severity = 2
    elif alarm.severity == 'moderate':
        alarm.severity = 1
    else:
        alarm.severity = 0

alarm_list = sorted(get_alarm_list),
                    key=lambda a: a.severity,
                    reverse=True)
return [alarms.Alarm.from_db_model(alarm)
        for alarm in alarm_list]

输出：

type is <type 'generator'>

列表中的对象是：

for alarm in get_alarm_list:
     print alarm

Output:
<aodh.storage.models.Alarm object at 0x7fa4c0cb1c50>
<aodh.storage.models.Alarm object at 0x7fa4c0cb17d0>
<aodh.storage.models.Alarm object at 0x7fa4c0d86f10>
<aodh.storage.models.Alarm object at 0x7fa4ca372110>
<aodh.storage.models.Alarm object at 0x7fa4ca372190>
<aodh.storage.models.Alarm object at 0x7fa4c0c55d90>

每个警报由以下数据组成

 {'alarm_actions': [u'log://'], 'ok_actions': [], 'description': u'instance running hot', 'state': u'insufficient data', 'fields': ['alarm_actions', 'ok_actions', 'severity', 'timestamp', 'description', 'time_constraints', 'enabled', 'state_timestamp', 'rule', 'alarm_id', 'state', 'insufficient_data_actions', 'repeat_actions', 'user_id', 'project_id', 'type', 'name'], 'repeat_actions': False, 'enabled': True, 'state_timestamp': datetime.datetime(2016, 5, 27, 6, 41, 5, 987428), 'rule': {u'meter_name': u'cpu_util', u'evaluation_periods': 3, u'period': 600, u'statistic': u'avg', u'threshold': 70.0, u'query': [], u'comparison_operator': u'gt', u'exclude_outliers': False}, 'name': u'ddd', 'alarm_id': u'f5045ed5-5c53-4a6e-be53-23d3368f40c6', 'time_constraints': [], 'insufficient_data_actions': [], 'timestamp': datetime.datetime(2016, 5, 27, 6, 41, 5, 987428), 'user_id': u'9a65b258b5a24e74ac5feae2f6c54229', 'project_id': u'28d1c27e782c4448bf53da00f49d3e1b', 'type': u'threshold', 'severity': 2}

如何迭代生成器？

alarm_list = sorted(get_alarm_list,
                    key=lambda a: a.severity,
                    reverse=True)

但是这里alarm_list 是空的。我如何在生成器get_alarm_list上使用排序功能@

Answer 1

您的代码中的问题是您正在尝试对耗尽的生成器进行排序（您在for 循环思想生成器之后进行排序）。您可以对生成器对象产生的结果进行排序，因此您可以选择将生成器耗尽到变量中，换句话说，基于get_alarm_list 创建新列表为list(generator)，然后使用简单的sorted 函数或@ 对其进行迭代和排序987654328@方法：

get_alarm_list = conn.query_alarms(query.filter_expr, query.orderb) 
sorted_alarm_list = sorted(list(get_alarm_list),
                               key=lambda a: a.severity,
                               reverse=True)
for alarm in sorted_alarm_list:
    print alarm

注意 1：list(get_alarm_list) 执行后 - get_alarm_list 生成器变为空。而唯一存储conn.query_alarms 结果的项目是sorted_get_alarm_list。您可以在 Generators Python Wiki 和 Understanding Generators in Python 上阅读有关生成器的更多信息

注意 2：实际上，您可以将生成器对象传递给 sorted，并且您将获得与传递 list(generator) 相同的列表，但是如果您将列表传递给 sorted，则它的工作速度更快（请参阅更多关于 SO 回答 sorted() using Generator Expressions Rather Than Lists)。

Answer 2

您遇到的问题是，当您修改对象的 severity 属性时，您正在使用 for 循环中的整个生成器。这意味着当您调用 sorted 时，没有什么可迭代的了，因为生成器只适合一种用途。

您可以通过摆脱第一个循环并将严重性转换逻辑放入 key lambda 函数来解决此问题：

alarms_gen = conn.query_alarms(query.filter_expr, query.orderb) 
alarms_list = sorted(alarms_gen,
                     key=lambda x: {'critical': -2, 'moderate': -1}.get(x.severity, 0))

请注意，我已重命名您的 get_alarms_list 变量以减少误导（它不是列表）。我还通过将较高的优先级映射到负键值，使排序调用中不需要reverse=True。