gpt4 book ai didi

Python numpy 数组对某些索引求和

转载 作者:太空狗 更新时间:2023-10-30 01:56:52 25 4
gpt4 key购买 nike

如何仅对 numpy 数组的索引列表执行求和,例如,如果我有一个数组 a = [1,2,3,4] 和要求和的索引列表, indices = [0, 2] 我想要一个快速操作来给我答案 4 因为 中索引 0 和索引 2 处的求和值的值a4

最佳答案

可以在用indices索引后直接使用sum:

a = np.array([1,2,3,4])
indices = [0, 2]
a[indices].sum()

关于Python numpy 数组对某些索引求和,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47734392/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com