Я пытаюсь отобразить элементы двоичной матрицы в список целых чисел. Необходимо l oop по каждой строке матрицы, если значение равно 1, то выбирается элемент из соответствующего списка, а если значение равно 0, элемент не выбирается. Конечная цель - получить вектор, содержащий сумму каждой строки соответствующих значений. Например:
listOfNums: [1,2,3,4,5]
m1= [[1,0,1,1,1]
[0,0,0,0,1]
[1,0,0,0,1]]
>>>[13,5,6]
Вот то, что я пробовал до сих пор, но я продолжаю получать ошибки индекса
def corrspondingVal(self, x):
nums = [1,2,3,4,5,6,7,8,9,10]
return [self.nums[i] for i in range(x) if x[i]]
def sumPerRow(self):
v = np.apply_along_axis(self.correspondingVals(self.matrix1), axis=1, arr=self.matrix1)
return v
(self.matrix1 - это (10,5) матрица двоичных значений )