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

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