Rewrote data_list to __iter__

This commit is contained in:
SimpleArt
2020-11-19 10:17:22 -05:00
parent 857e248034
commit 6e95ff5d9d
3 changed files with 25 additions and 22 deletions

View File

@ -53,9 +53,11 @@ class Chromosome:
"""Set the fitness value of the chromosome"""
self.fitness = fitness
def data_list(self):
"""Returns a list of the gene values"""
return [gene.value for gene in self.gene_list]
def __iter__(self):
"""Returns an iterable of the gene values"""
for gene in self.gene_list:
yeild gene.value
def __repr__(self):

View File

@ -149,9 +149,10 @@ class Population:
self.fitness = fitness
def data_list(self):
"""Returns a list of chromosome data lists"""
return [chromosome.data_list() for chromosome in self.chromosome_list]
def __iter__(self):
"""Returns an iterable of chromosome iterables"""
for chromosome in self.chromosome_list:
yield iter(chromosome)
def __repr__(self):