将函数的多个返回值分配给列表中实例的属性

时间:2017-04-12 08:50:31

标签: python class multiple-assignment

我有一个类实例列表Child和我想使用从函数返回的玩具列表来修改每个孩子的玩具属性

下面有一个可行的解决方案,但我想知道是否有单行?

import random

class Child():
    def __init__(self, name, toy):
        self.name = name
        self.toy = toy

def change_toys(nChildren):
    toys = ['toy1', 'toy2', 'toy3', 'toy4']
    random.shuffle(toys)
    return toys[:nChildren]

child_list = [Child('Ngolo', None), Child('Kante', None)]

new_toys = change_toys(len(child_list))

for i in range(len(child_list)):
    child_list[i].toy = new_toys[i]
    print "%s has toy %s" %(child_list[i].name, child_list[i].toy)

输出(随机玩具分配):

Ngolo has toy toy3
Kante has toy toy2

我试过了:

[child.toy for child in child_list] = change_toys(nChildren)

但那不起作用,我得到

SyntaxError: can't assign to list comprehension

有什么想法吗?

2 个答案:

答案 0 :(得分:0)

这不是一个单行,但你应该通过避免使用你的索引i以更干净,更py的方式编写你的循环:

for child, toy in zip(child_list, new_toys):
    child.toy = toy
    print "%s has toy %s" %(child.name, child.toy)

答案 1 :(得分:0)

如果确实想要使用列表解析,则应创建一个新列表来创建新的Child对象,而不是改变现有对象:

In [6]: new_list = [Child(c.name, toy) for c, toy in zip(child_list, change_toys(len(child_list)))]

In [7]: new_list
Out[7]: [<__main__.Child at 0x103ade208>, <__main__.Child at 0x103ade1d0>]

In [8]: c1, c2 = new_list

In [9]: c1.name, c1.toy, c2.name, c2.toy
Out[9]: ('Ngolo', 'toy4', 'Kante', 'toy2')

与原始清单相比:

In [10]: c1, c2 = child_list

In [11]: c1.name, c1.toy, c2.name, c2.toy
Out[11]: ('Ngolo', None, 'Kante', None)

如果您更愿意改变您的子实例(一种合理的方法),那么您的for循环就可以了。

我会像@ThierryLathuille一样使用zip进行for循环。