Lista zapełnień z krotkami

Question

Po prostu bawię się symulacją (Mendel's First Law of Inheritance).

Zanim mogę pozwolić współmieszkom zwierząt i przeanalizować wynik, należy wygenerować populację, tj. Lista musi być wypełniona zmienną liczbą trzech różnych typów krotek bez rozpakowywania ich.

Starając się zapoznać z itertools (Muszę kombinacje później w części współpracującej), wymyśliłem następujące rozwiązanie:

import itertools 

k = 2 
m = 3 
n = 4 

hd = ('A', 'A')  # homozygous dominant 
het = ('A', 'a')  # heterozygous 
hr = ('a', 'a')  # homozygous recessive 

fhd = itertools.repeat(hd, k) 
fhet = itertools.repeat(het, m) 
fhr = itertools.repeat(hr, n) 

population = [x for x in fhd] + [x for x in fhet] + [x for x in fhr] 

co skutkowałoby:

[('A', 'A'), ('A', 'A'), ('A', 'a'), ('A', 'a'), ('A', 'a'), ('A', 'a'), ('A', 'a'), ('A', 'a'), ('A', 'a')] 

Czy istnieje rozsądniejszy, pytonowy lub oszczędzający pamięć sposób na zbudowanie ostatecznej listy, np. bez generowania najpierw list dla trzech typów osób?

Answer 1

Można użyć itertools.chain połączyć iteratorów:

population = list(itertools.chain(fhd, fhet, fhr)) 

Chociaż powiedziałbym, że nie ma potrzeby korzystania itertools.repeat kiedy można po prostu zrobić [hd] * k. Rzeczywiście, chciałbym podejść do tej symulacji następująco:

pops = (20, 30, 44) 
alleles = (('A', 'A'), ('A', 'a'), ('a', 'a')) 

population = [a for n, a in zip(pops, alleles) for _ in range(n)] 

czy może

allele_freqs = ((20, ('A', 'A')), 
       (30, ('A', 'a')), 
       (44, ('a', 'a'))) 

population = [a for n, a in allele_freqs for _ in range(n)] 

Answer 2

to powinno działać jak sądzę.

pops = [2,3,4] 
alleles = [('A','A'), ('A', 'a'), ('a','a')] 
out = [pop*[allele] for pop, allele in zip(pops,alleles)] 
print [item for sublist in out for item in sublist] 

Umieściłem kod na CodeBunk, aby można było go uruchomić.

Answer 3

population = 2*[('A', 'A')] + 3*[('A', 'a')] + 4*[('a', 'a')] 

lub

hd = ('A', 'A')  # homozygous dominant 
het = ('A', 'a')  # heterozygous 
hr = ('a', 'a')  # homozygous recessive 

population = 2*[hd] + 3*[het] + 4*[hr]