时间戳

首页 > 解决方案 > 如何添加列表索引给出的数字？

python-3.x - 如何添加列表索引给出的数字？

问题描述

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

标签： python-3.xloopsaddition

解决方案

因此，如果我做对了，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]]

推荐阅读