Python-Array aus dem ersten Element von Array 1, dem ersten Element von Array 2, dem zweiten Element von Array 1, dem zweiten Element von Array 2 usw. erstellen

Question

Ich versuche einen Zustandsvektor zu erstellen, der die Positionen und Geschwindigkeiten einer Reihe von Partikeln darstellt zu einer bestimmten Zeit für eine Simulation. Ich habe individuelle Vektoren x, y, vx, vy erstellt, die den Wert dieser Variablen für jedes Partikel angeben. Gibt es eine gute Möglichkeit, sie automatisch zu einem Array zu kombinieren, das alle Informationen für Partikel eins enthält, gefolgt von allen Informationen für Partikel zwei usw.? Danke

Answer 1

Meinst du das?

x = [0, 1, 2] 
y = [3, 4, 5] 
vx = [6, 7, 8] 
vy = [9, 10, 11] 

c = zip(x, y, vx, vy) 
print(c) # -> [(0, 3, 6, 9), (1, 4, 7, 10), (2, 5, 8, 11)] 

wenn Sie mit Python 3, müssen Sie c = list(zip(x, y, vx, vy)) verwenden.

Wenn Sie nicht die Werte für jeden in ein Tupel gruppiert Teilchen wollen wie das aus irgendeinem Grund, könnte das Ergebnis abgeflacht werden:

c = [item for group in zip(x, y, vx, vy) for item in group] 
print(c) # -> [0, 3, 6, 9, 1, 4, 7, 10, 2, 5, 8, 11] 

jedoch, würde ich empfehlen, nur „Benennung“ der stattdessen Tupel:

from collections import namedtuple 

Particle = namedtuple('Particle', 'x, y, vx, vy') 
c = [Particle._make(group) for group in zip(x, y, vx, vy)] 
print(c) 

Ausgang:

[Particle(x=0, y=3, vx=6, vy=9), 
Particle(x=1, y=4, vx=7, vy=10), 
Particle(x=2, y=5, vx=8, vy=11)] 

Auf diese Weise können Sie die Felder nach Namen referenzieren - d. H. c[1].x - was den nachfolgenden Code und die Berechnungen viel lesbarer machen könnte.