我创建了一个创建多维列表的类。我还定义了一个方法来获取此列表中的特定值;但是,我希望能够使用列表切片获取特定值。
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]]
是否有可能在没有迭代的情况下得到这个?