Python - 从列表中提取并 reshape

Question

我有一个包含 no_classes 元素的列表，其中每个元素都是形状为 no_people x 2 的 numpy 数组。

例如，这个名为 scores 的列表是使用

构建的

no_classes = 3
no_people = 2

我希望从中提取每个人的第二个元素(始终只是第二个元素)，并将其 reshape 为形状为 no_people x no_classes

的矩阵

scores=[np.array([[1,2],[3,4]]),np.array([[5,6],[7,8]]),np.array([[9,10],[11,12]])]

对于分数，这将产生以下矩阵:

有什么最好的方法吗？

Answer 1

这是方法之一:

a=[x[:,1] for x in scores]

for i,el in enumerate(a):
    if i ==0:
        hold=el
    else:
        hold=np.column_stack([hold,el])