整数列表中特定元素的总和

Question

我必须为倦怠问卷编写一个脚本，在脚本的末尾会给我三个分量表的分数。 我定义了一个包含所有 22 个问卷问题的列表和一个存储所有答案的列表。 现在，我有一个包含 22 个值（RatingScale，low = 0，high = 6）的答案列表（称为Answerlist ）。 使用 numpy 库，我知道如何对所有值求和

total = numpy.sum(Answerlist)

我不知道如何 select 每个分量表的相关项目/问题。 我试过了

subscale1 = numpy.sum(Answerlist[:, [0, 1, 2, 5, 7, 12, 13, 15, 19]])

subscale2 = numpy.sum(Answerlist[:, [4, 9, 10, 14, 21]]) 

但我收到错误TypeError: list indices must be integers or slices, not tuple 。

Answer 1

如果Answerlist是一维数组，那么您将按如下方式对其进行索引

subscale1 = numpy.sum(Answerlist[[0, 1, 2, 5, 7, 12, 13, 15, 19]])

subscale2 = numpy.sum(Answerlist[[4, 9, 10, 14, 21]]) 

例如

>>> data = np.random.randint(0,6,22)
>>> data
array([2, 4, 1, 0, 2, 0, 2, 1, 3, 4, 1, 2, 3, 2, 1, 0, 1, 0, 5, 3, 3, 1])
>>> data[[1,12,18]]
array([4, 3, 5])
>>> np.sum(data[[1,12,18]])
12