如何添加列表索引中给出的数字？

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