Export nach Numpy

Das Ergebnis einer Abfrage lässt sich mit der Funktion fetchnumpy() in ein Numpy-Array umwandeln. Zum Beispiel:

import duckdb
import numpy as np

my_arr = duckdb.sql("SELECT unnest([1, 2, 3]) AS x, 5.0 AS y").fetchnumpy()
my_arr

{'x': array([1, 2, 3], dtype=int32), 'y': masked_array(data=[5.0, 5.0, 5.0],
             mask=[False, False, False],
       fill_value=1e+20)}

Anschließend kann das Array mit Numpy-Funktionen weiterverarbeitet werden, z. B.:

np.sum(my_arr['x'])

Siehe auch

DuckDB unterstützt auch den Import aus Numpy.