File size: 316 Bytes
4a51346
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
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)