python - Python - 从列表中提取并重塑

Question

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

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

no_classes = 3
no_people = 2

我希望从中提取每个人的第二个元素（始终只是第二个）并重塑为形状no_peoplex的矩阵no_classes

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

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

任何想法如何最好地接近？

标签： python

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])