根據位置將嵌套列表中的元素提取到單獨的列表中

Question

我有一個嵌套列表說：

[[1,2],
 [3,4],
 [5,6]]

如何從該列表中提取一列（ [1,3,5]或[2,4,6] ）而不將其轉換為pandas DataFrame或np array ，例如：

a = [[1,2],[3,4],[5,6]]
list(pd.DataFrame(a)[0])

或者

a = [[1,2],[3,4],[5,6]]
list(np.array(a)[:,0])

兩者都產生[1,3,5] 。

Answer 1

像這樣

col = 0
[item[col] for item in a]

Answer 2

使用zip將您的列表解壓縮為：

a = [[1,2],[3,4],[5,6]]

list1, list2 = zip(*a)

zip返回一個元組迭代器，然后將其解包到list1和list2中，如果您想要列表， map list然后解包：

list1, list2 = map(list,zip(*a))
print(list1)
[1, 3, 5]
print(list2)
[2, 4, 6]

zip聚合來自輸入迭代的元素。 通過使用zip(*a)解包，我們使每個內部列表在zip中成為一個單獨的可迭代對象，然后根據它們的位置“組合”每個內部列表中的元素。

Answer 3

你可以試試map function 。

y=list(map(lambda x:x[0],a))

Answer 4

如果您想要one-line list comprehension ，這可能會有所幫助：

a = [[1,2],[3,4],[5,6]]

print([[elem[i] for elem in a] for i in range(len(a[0]))])

注意： len(a[0])是我正在使用的，因此這僅適用inner-lists of equal lengths 。

Answer 5

我是 python 的初學者，請你試試這個。

a = [[1,2],[3,4],[5,6]]
for i in a:
    print(i[0])