Wie wandelt man ein pandas DataFrame in ein NumPy Array um?

In pandas kann man ein DataFrame in ein NumPy-Array umwandeln, indem man das values-Attribut verwendet.

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3],
                  'B': [4, 5, 6]})

numpy_array = df.values

Dies gibt ein 2-dimensionales NumPy-Array zurück, das alle Daten im DataFrame enthält. Die Form des Arrays ist (n, m), wobei n die Anzahl der Zeilen und m die Anzahl der Spalten im DataFrame ist.

Alternativ kann man auch die Methode to_numpy() verwenden, die ebenfalls ein NumPy-Array mit den Daten im DataFrame zurückgibt.

numpy_array = df.to_numpy()

Man kann auch bestimmte Spalten des DataFrames in ein NumPy-Array umwandeln, indem man den Spaltennamen an das values-Attribut oder die to_numpy() Methode übergibt

numpy_array = df['A'].values
numpy_array = df[['A','B']].to_numpy()

Bitte beachten Sie, dass das resultierende NumPy-Array nicht die gleichen Spaltennamen wie das ursprüngliche DataFrame hat, falls Sie die Spaltennamen benötigen, können Sie das .columns Attribut des DataFrames verwenden.