python - 如何在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 对其进行迭代和排序函数或 list.sort 方法:

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 answer sorted() using Generator Expressions Rather Than Lists 的更多信息)。