Блог

У меня есть список, который содержит no_classes элементы, где каждый элемент представляет собой числовой массив формы no_people x 2.

В качестве примера, этот вызываемый список scores создается с

 no_classes = 3
no_people = 2
 

Я хочу извлечь из этого второй элемент для каждого пользователя (всегда только второй) и преобразовать в матрицу формы no_people x no_classes

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

Для оценок это приведет к следующей матрице:

Есть идеи, как лучше подойти?

Это один из подходов:

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