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

Question

我必须为一份职业倦怠问卷写一个脚本，在脚本的最后会给出三个分量表的分数。我定义了一个包含调查问卷的所有22个问题的列表和一个存储所有答案的列表。现在，我有了包含22个值(RatingScale，low = 0，high = 6)的答案列表(称为Answerlist)。使用numpy库，我知道如何对所有值求和。

total = numpy.sum(Answerlist)

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

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