我有一个类实例列表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
有什么想法吗?
答案 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循环。