⌘+k ctrl+k
1.3 (稳定版)
搜索快捷键 cmd + k | ctrl + k
导出到 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

另请参阅

DuckDB 还支持从 Numpy 导入