设置已定义的类方法以接受列表

时间:2015-01-24 05:50:16

标签: python python-3.x slice

我创建了一个创建多维列表的类。我还定义了一个方法来获取此列表中的特定值;但是,我希望能够使用列表切片获取特定值。

class Agents():
    def __init__(self, num_agents):
    a = [['a'],[2],['Hello']]
    self.agent_array = [a[:] for i in range(num_agents)] 

上面的代码根据num_agents号码创建列表a的副本。我想要做的是只能获得a所有副本的第二列,即

def _getcolumn(self, agent_index):
    print(self.agent_array[agent_index][1])

_getcolumn只接受整数。如何更改它以接受切片或范围以获取多维列表的整个第二列的值?

a = Agents(2)
b = a._getcolumn([0:2])
b = [[2],[2],[2]]

是否有可能在没有迭代的情况下得到这个?

0 个答案:

没有答案
相关问题