如何制作一个新列表，该列表具有每个列表的第一个元素但分为 3 个？

Question

如果我从这个开始

A = [[1,3,5,7,2,2], [2,3,1,2,1,1]]
B = [[3,2,5,5,6,6], [2,1,4,2,7,2]]
C = [[1,0,2,4,8,2], [9,7,2,2,7,2]]

我怎样才能实现这个 output？

[[1,3],[5,7],[2,2],[3,2],[5,5],[6,6],[1,0],[2,4],[8,2]] 

Answer 1

如果它是列表，你可以做一个嵌套的 for 循环：

result = []
for list_ in [A,B,C]:
    for i in range(int(len(list_[0])/2)):
        result.append(list_[0][i*2:i*2+2])

[[1, 3], [5, 7], [2, 2], [3, 2], [5, 5], [6, 6], [1, 0], [2, 4], [8, 2]]

或者列表理解

[list_[0][i*2:i*2+2] for list_ in [A,B,C] for i in range(int(len(list_[0])/2))]

如果是 numpy 数组，可以使用 concatenate 和 reshape

concat_array = np.concatenate((A[0],B[0],C[0]))

concat_array.reshape(9,2)

array([[1, 3],
       [5, 7],
       [2, 2],
       [3, 2],
       [5, 5],
       [6, 6],
       [1, 0],
       [2, 4],
       [8, 2]])

Answer 2

numpy 爱好者不会喜欢它，但您可以提取第一个元素作为常规 python 列表并加入它们（下面的first_elems ），然后从该列表中取出成对元素并转换回 numpy 数组。

from numpy import array
A = array([[1,3,5,7,2,2], [2,3,1,2,1,1]])
B = array([[3,2,5,5,6,6], [2,1,4,2,7,2]])
C = array([[1,0,2,4,8,2], [9,7,2,2,7,2]])

first_elems = list(A[0]) + list(B[0]) + list(C[0])
array(list(zip(first_elems[::2], first_elems[1::2])))