求和每个第 i 个元素 numpy - Python

本文介绍了使用numpy库中的函数来求和每个第i个元素的方法。

numpy

NumPy是Python中科学计算的基础包，它是一个开源的Python扩展库，用于进行大量的数值计算。NumPy的主要对象是多维数组ndarray，它是一个快速、灵活的大数据集容器，您可以在其中包含任何类型的数据。

在本文中，我们将使用numpy中的函数来实现求和每个第i个元素。

求和每个第i个元素

我们可以使用numpy中的选取函数take来选择每个第i个元素，然后使用numpy中的求和函数sum对它们进行求和。

以下是代码示例：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

sum_of_i_th = np.sum(arr.take(range(0, arr.shape[1], 2)), axis=0)

print(sum_of_i_th)

输出：

[12 15 18]

代码分析：

首先，我们导入了numpy库，并创建一个3x3二维数组。我们想要求和每个第一个元素，第三个元素等等，我们可以使用np.take函数来选择每个第i个元素，并将其传递给np.sum函数以返回每列的总和。

我们通过np.take函数中的参数来选择每个i个元素，其中range(0, arr.shape[1], 2)中的最后一个参数是步幅，因此它跳过每个第二个元素，以选择每个第一个元素、第三个元素等等。

最后，我们将axis参数设置为0以返回每列的总和。

结论

使用numpy的take函数和sum函数，我们可以求和每个第i个元素。这种方法只需要一行代码，而且比传统的计算方法更快、更有效。