從python中獲取Numpy數組的SQL頭文件

Question

使用下面的代碼，我可以從SQL中獲取行和列數據：
如何獲取表頭作爲結果集或數組的一部分。

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

就像我在這裏問： How do I create a CSV file from database in Python?

Answer 1

如果你想要的列名數組中的第一行，你會做

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) 

Answer 2

csr.description應該有頭

Answer 3

這是一個自包含的例子，說明了總體思路。 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')]