如何只在 python 中添加数组的第一个元素?

How can I add only the first elements of an array in python?

如果我有一个数组,例如:a = np.array([[1,2], [2,5], [4,8]]),其中第一个数字代表x值,第二个代表 y 值,我如何添加所有第一个数字(x 值)和所有第二个数字(y 值)。 输出应如下所示: 7, 15 谢谢!

a = np.array([[1,2], [2,5], [4,8]])
x = 0
y = 0
for element in a:
    x += element[0]
    y += element[1]
print("x: " + str(x))
print("y: " + str(y))

输出:

>> x: 7
>> y: 15

访问数组的每个元素,然后对于每个元素,将第一个元素作为 x,将第二个元素作为 y

x, y = a[:,0].sum(), a[:,1].sum()

您可以对行轴(轴 = 0)使用 numpy sum()。这对每行相同位置的值求和,或者换句话说,它对同一列中的值求和。

import numpy as np
a = np.array([[1,2], [2,5], [4,8]])
print(np.sum(a,axis=0))

使用 numpy 内置函数

np.sum(a,axis=0)