主页 / 问题 / 630258
Accepted
Алексей Воронов
Asked:2020-02-20 18:21:59 +0000 UTC2020-02-20 18:21:59 +0000 UTC

将数组的多个元素相加

  • 772

有一个数组numpy。如何在不使用循环的情况下将数组的所有元素或选择性元素相加，例如从 5 到 100？

python

1 个回答

  1. Best Answer

    例子：

    In [83]: import numpy as np

In [84]: a = np.random.randint(0, 10, 15)

In [85]: a
Out[85]: array([7, 8, 8, 6, 4, 0, 3, 0, 4, 0, 8, 0, 1, 4, 5])

    对索引为 5 到 10 的元素求和：

    In [86]: a[5:10]
Out[86]: array([0, 3, 0, 4, 0])

In [87]: a[5:10].sum()
Out[87]: 7

    对值从 5 到 7 的元素求和：

    In [88]: a[(a >= 5) & (a <= 7)]
Out[88]: array([7, 6, 5])

In [89]: a[(a >= 5) & (a <= 7)].sum()
Out[89]: 18

    PS 添加所有元素：

    In [105]: a.sum()
Out[105]: 58
    • 4

相关问题