np.column_stack() Verwendung innerhalb der Schleife

Question

Ich frage mich, ob es eine Möglichkeit gibt, diesen Code arbeiten zu lassen. Hoffentlich ist klar, was ich machen möchte.

Ich hoffte, dass nach jeder Iteration eine neue Zeile innerhalb des Arrays erstellt würde. Stattdessen scheint es so, als ob die vorherige Zeile überschrieben wurde - nicht überraschend.

Gibt es eine andere Möglichkeit, dies zu tun?

list1 = [4, 5] 

for j in list1: 

      one = j + 1 
      two = j + 2 
      array = np.column_stack((one, two)) 

print(array) 

#Output 
[[ 6 7]] 

#What I actually want 
[[ 5 6] 
[ 6 7]] 

Answer 1

Die natürliche Art und Weise würde dies in Python zu tun:

import numpy 

list1 = [4,5] 
array = numpy.column_stack((j+1,j+2) for j in list1) 

print(array) 

Einige Leute Kodierung als pythonic auf diese Art und Weise beziehen, während der Annäherung an ähnlicher ist, was man in C tun würde ++ oder Java . Sie werden sich daran gewöhnen, wenn Sie weiterhin mit Python arbeiten.

Wenn Sie die Dokumentation von numpy.colum_stack lesen, werden Sie sehen, dass das Funktionsargument ein Tupel von Spalten ist, die in ein 2d-Array "gestapelt" werden. dieses Tupel als

[(j+1,j+2) for j in list1] 

(was auch eine gültige Zeile von Python-Code) erzeugt werden. Innerhalb eines Funktionsarguments können wir die eckigen Klammern weglassen.

Ihr Beispiel funktioniert nicht, weil Sie Array in jeder einzelnen Schleifeniteration überschreiben. Ich empfehle, einen Debugger zu verwenden ... oder einen Ausdruck (Array) innerhalb der for-Schleife hinzuzufügen, um besser zu verstehen, was vor sich geht.