如何让for循环进入列表理解？

Question

我的 for 循环如下：

Object_id_lst = []

for id_tic in ticket_df['ID']:
    get_object_id = api_client.crm.tickets.associations_api.get_all(ticket_id=id_tic, to_object_type='contacts')
    if len(get_object_id.results) > 0:
        Object_id_lst.append(get_object_id.results[0].to_object_id)
    else:
        object_id = 0
        Object_id_lst.append(object_id)

我想看看是否有办法将其纳入列表推导式以缩短计算时间。

Answer 1

我认为将其作为列表理解不会有多大帮助，其他优化可能会有更多帮助。 我做了两个：

Object_id_lst = [
    results[0].to_object_id if results else 0
    for get_all in [api_client.crm.tickets.associations_api.get_all]
    for id_tic in ticket_df['ID']
    for results in [get_all(ticket_id=id_tic, to_object_type='contacts').results]
]

我的优化（除了使它成为一种理解）：

• 只获取一次get_all function 而不是每次。
• 只获取.results一次而不是两次。
• 检查results的真实性而不是对其调用len ，与0进行比较，并检查比较结果的真实性。

另一种选择（应该在 function 中，而不是全局的，因此您可以获得更快的变量）：

Object_id_lst = []
append = Object_id_lst.append
get_all = api_client.crm.tickets.associations_api.get_all
for id_tic in ticket_df['ID']:
    for result in get_all(ticket_id=id_tic, to_object_type='contacts').results:
        append(result.to_object_id)
        break
    else:
        append(0)

Answer 2

这不是很漂亮，但您可以通过使用海象运算符在列表推导中分配get_object_id来实现，这样您就可以测试它的长度并在推导的值部分使用它。

Object_id_list = [
    get_object_id.results[0].to_object_id 
        if len((get_object_id := api_client.crm.tickets.associations_api.get_all(ticket_id=id_tic, to_object_type='contacts')).results) > 0 
        else 0
    for id_tic in ticket_df['ID']]

更现实地说，我会将所有条件代码移动到 function 和 map 中。

def object_id(id_tic):
    get_object_id = api_client.crm.tickets.associations_api.get_all(ticket_id=id_tic, to_object_type='contacts')
    if len(get_object_id.results) > 0:
        return get_object_id.results[0].to_object_id
    else:
        return 0

Object_id_list = list(map(object_id, ticket_df['ID']))

Answer 3

Object_id_list = [api_client.crm.tickets.associations_api.get_all(ticket_id=id_tic, to_object_type='contacts').results[0].to_object_id 
                  if len(api_client.crm.tickets.associations_api.get_all(ticket_id=id_tic, to_object_type='contacts')) > 0
                  else 0
                  for id_tic in ticket_df['ID']]