Как добавить числа по индексу списка?

Question

У меня есть X_train, который равен [1000,2], а pos_list - это список индекса, который мне нужно добавить.То есть pos_list = [2,58,78, ...], который обозначает строки x_train, которые необходимо сложить вместе.Так что мои sum_pos должны быть [1,2]

for i in range(len(pos_list)):
    sum_pos = sum(X_train(pos_list[i])

Answer 1

Так что, если я правильно понял, X_train - это матрица / таблица с 1000 строками и 2 столбцами, и вам нужно добавить строки, заданные pos_list.Ваша формулировка очень запутанная, но, если я правильно понимаю, такое решение может сработать:

X_train = ... # Our matrix
sum_pos = 0
for index in pos_list:
    sum_pos += sum(X_train[index])

ИЛИ однострочная версия (хотя и не такая красивая):

sum_pos = [sum(X_train[index]) for index in [pos for pos in pos_list]]