Python3.5中列表索引的快捷方式是什么?

时间:2018-01-29 15:33:26

标签: python pandas indexing

我想获取列名,以便我可以添加到新数据框。

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()来实现这一目标?

1 个答案:

答案 0 :(得分:1)

这可能有用。它并不比你拥有的好多了。

columns = list(zip(*cursor.description))[0]