从Python中的Numpy数组获取SQL头

Question

通过下面的内容，我可以从 SQL 中获取行和列数据：
如何获取表头作为结果集或数组的一部分？

    top = csr.execute("Select * from bigtop")
    d=list(top)
    a = np.asarray(d, dtype='object')
    print a

就像我在这里问的： 如何从数据库创建 CSV 文件Python？

Answer 1

这是一个独立的示例，说明了总体思路。 numpy.recarray 是你的朋友，

from sqlite3 import connect
from numpy import asarray

db = connect(":memory:")
c = db.cursor()
c.execute('create table bigtop (a int, b int, c int)')

for v in [(1,2,3),(4,5,6),(7,8,9)]:
    c.execute('insert into bigtop values (?,?,?)',v)

s = c.execute('select * from bigtop')

h = [(i[0],int) for i in c.description]

# You can also use 'object' for your type
# h = [(i[0],object) for i in c.description]

a = asarray(list(s),dtype=h)

print a['a']

给出第一列，

[1 4 7]

并

print a.dtype

给出每列的名称和类型，

[('a', '<i4'), ('b', '<i4'), ('c', '<i4')]

或者，如果你使用 object 作为你的类型，你会得到，

[('a', '|O4'), ('b', '|O4'), ('c', '|O4')]

Answer 2

csr.description 应该有标题

Answer 3

如果您希望列名称作为数组中的第一行，您会

top = csr.execute("Select * from bigtop")
d=list(top)
a = np.asarray([[x[0] for x in top.description]] + d, dtype='object')

得到类似的结果

array([[heading1, heading2, heading3, ...],
       [val1, val2, val3, ...],
           ...
           , dtype=object)