⌘+k ctrl+k
1.1.3 (stable)
Search Shortcut cmd + k | ctrl + k
Export to Numpy

查询的结果可以使用fetchnumpy()函数转换为Numpy数组。例如:

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)}

然后,可以使用Numpy函数处理数组,例如:

np.sum(my_arr['x'])
6

See Also

DuckDB 还支持 从 Numpy 导入