有没有一种方法可以通过只知道实例数组中实例的属性来获取该实例的索引?
我有这样的东西:
class NodeGene:
def __init__(self):
self.__initBias()
def __initBias(self):
#Creates a random bias and assigns it to self.bias
class ConnectionGene:
#nodeIn and nodeOut are instances of NodeGene
def __init__(self, nodeIn, nodeOut, innovationNumber):
self.nodeIn = nodeIn
self.nodeOut = nodeOut
self.innovationNumber = innovationNumber
self.__initWeight()
def __initWeight(self):
#Creates a random weight and assigns it to self.weight
class Genome:
def __init__(self, connections):
#connections is an array of ConnectionGene instances
self.connections = connections
如果我有要查找的实例的nodeIn和innovationNumber,我如何在连接中获得ConnectionGene的索引?
假设conn_list
是ConnectionGene
实例的列表。然后你有几个选择:
idx = None
for i, c in enumerate(conn_list):
if c.nodeIn == 0 and c.innovationNumber == 0:
idx = i
break
或
idx_list = [i for i, c in enumerate(conn_list) if c.nodeIn == 0 and c.innovationNumber == 0]
或
idx = next(i for i, c in enumerate(conn_list) if c.nodeIn == 0 and c.innovationNumber == 0)
如果你要多次这样做,最好制作一个参考字典并在那里进行快速查找:
dct = {(c.nodeIn, c.innovationNumber): i for i, c in enumerate(conn_list)}
...
idx = dct[0, 0] # very fast
以下是实现这一点的一种方法。我不确定你想把你的代码调用到哪里,所以我只把连接称为";连接";在下面
indices = [index for index, elem in enumerate(connections) if elem.nodeIn == ___ if elem.innovationNumber == ____]
if indices:
return indices[0]
return -1
只需填空即可。显然,您可以更改是返回第一个索引还是只返回所有索引。
如果要检查nodeIn是否与另一个NodeGene是同一个对象实例,可以使用is
而不是==
。如果使用==
,则可以在NodeGene类上定义__eq__
方法。