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

时间:2019-02-21 17:05:44

标签: python-3.x loops addition

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

1 个答案:

答案 0 :(得分: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]]