blob: df99550d348a89dd4086050358591ac94ad50467 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
|
from clickhouse_connect.driver.options import np
from clickhouse_connect.driver.types import ByteSource
def read_numpy_array(source: ByteSource, np_type: str, num_rows: int):
dtype = np.dtype(np_type)
buffer = source.read_bytes(dtype.itemsize * num_rows)
return np.frombuffer(buffer, dtype, num_rows)
|
