Python - 如何獲取每個列表的第一行？

Question

我需要獲取列表的每一行並創建一個新列表。 讓我解釋。

我有這個：

data = [[204.0, u'stock'], [204.0, u'stock']]

我需要改變這個：

column1 = [204.0, 204.0]
colunm2 = [u'stock', u'stock']

關於如何做到這一點的任何線索？

最好的祝福，

Answer 1

使用zip() ：

>>> data = [[204.0, u'stock'], [204.0, u'stock']]
>>> zip(*data)
[(204.0, 204.0), (u'stock', u'stock')]
>>> column1, column2 = zip(*data)
>>> column1
(204.0, 204.0)
>>> column2
(u'stock', u'stock')

或者，來自itertools izip() ：

>>> from itertools import izip
>>> column1, column2 = izip(*data)
>>> column1
(204.0, 204.0)
>>> column2
(u'stock', u'stock')

Answer 2

一個簡單的列表理解就可以了。

data = [[204.0, u'stock'], [204.0, u'stock']]

column1 = [i[0] for i in data]
column2 = [i[1] for i in data]

>>> column1
 [204.0, 204.0]
>>> column2
 ['stock', 'stock']

Answer 3

可能是，在一行

>>> data = [[204.0, u'stock'], [204.0, u'stock']]
>>> columns = [ [d[k] for d in data] for k in range(2)]
[[204.0, 204.0], ['stock', 'stock']]
>>> columns[0]
[204.0, 204.0]

如果變量的大小可以改變，你可以這樣做：

columns = [ [d[k] for d in data] for k in range(max(map(len,data)))]