python - Kdb 数据库到 PyQ 中的 NumPy 数组

Question

我有一个展开的 Kdb 数据库，包含符号、 float 和时间戳。我想将其转换为 NumPy 数组。但是使用下面的代码...

>>> import numpy as np
>>> from pyq import q
>>> d = q.load(':alpha/HDB/')
>>> a = np.array(d)

返回这个错误...

Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/Users/marrowgari/py3/lib/python3.6/site-packages/pyq/_n.py", line 158, in array
return numpy.array(list(self), dtype)
TypeError: iteration over a K scalar, t=-11

这是因为 Kdb 符号类型在 NumPy 中没有直接的对应物吗？如果是这样，我该如何纠正？

Answer 1

假设您的 HDB 是这样创建的:

q)(` sv db,`t`)set .Q.en[db:`:alpha/HDB]([]sym:`A`B`C;a:1 2 3)
`:alpha/HDB/t/
q)\l alpha/HDB
q)t
sym a
-----
A   1
B   2
C   3

然后，首先你应该使用 \l 命令加载它，而不是加载函数:

>>> q('\\l alpha/HDB')
k('::')

这将加载您所有的表和枚举域。

现在您应该能够将表的 sym 列转换为 numpy 字符串数组

>>> np.array(q.t.sym)
array(['A', 'B', 'C'], dtype=object)

或者一个 numpy 整数数组:

>>> np.array(q.t.sym.data)
array([0, 1, 2])

您也可以一次性将整个表转换为一个 numpy 记录数组，但您必须先将其“映射”到内存中:

>>> np.array(q.t.select())
array([('A', 1), ('B', 2), ('C', 3)], dtype=[('sym', 'O'), ('a', '<i8')])