我想获取列名,以便我可以添加到新数据框。
In [241]: cursor.description
Out[241]:
(('a', None, None, None, None, None, None),
('b', None, None, None, None, None, None),
('c', None, None, None, None, None, None),
('d', None, None, None, None, None, None))
In [242]: rows
Out[242]:
[('Atlanta', 'Georgia', 1.25, 6),
('Tallahassee', 'Florida', 2.6, 3),
('Sacramento', 'California', 1.7, 5)]
我使用了相当复杂和沉闷的方法:
frame = DataFrame(rows, columns = list(cursor.description[i][0] for i in range(len(cursor.description))))
是否有一个捷径来调用列名?是否可以通过使用zip()来实现这一目标?
答案 0 :(得分:1)
这可能有用。它并不比你拥有的好多了。
columns = list(zip(*cursor.description))[0]