如何在numpy中添加两个数组

Question

这是一个示例代码，我想添加两个数组的元素。我已经导入了 NumPy，不想导入数组。

from numpy import *

a = array([])
b = array([])
c = array([])

d = input("Enter the length of the arrays")
print ("Enter the elements of array 1")
for i in range(d):
    append(a, int(input("Enter the element ")))
print ("Enter the elements of array 2")
for i in range(d):
    append(b, int(input("Enter the element ")))

for i in range(1, d+1):
    append(c, (a[i] + b[i]))

print(a)
print(b)
print(c)

预期的输出应该是数组元素的总和，但我得到以下错误：

IndexError：索引 1 超出轴 0 的范围，大小为 0

Answer 1

首先您需要将d 转换为int：

d = int(input("Enter the length of the arrays"))

之后，您应该将append 函数的结果分配给array：

print ("Enter the elements of array 1")
for i in range(d):
    a = append(a, int(input("Enter the element ")))
print ("Enter the elements of array 2")
for i in range(d):
    b = append(b, int(input("Enter the element ")))

那么结果就好了

Answer 2

函数append 不会改变其参数的状态。相反，它返回添加了元素的数组。

所以要将元素添加到数组中，您需要：a = append(a, 2)

同样在 NumPy 中，您可以使用 add(a,b) 对两个数组求和。

如果我们将所有这些应用到您的示例中，我们会得到：

from numpy import *

a = array([])
b = array([])
c = array([])

d = input("Enter the length of the arrays")
print ("Enter the elements of array 1")
for i in range(d):
    a = append(a, int(input("Enter the element ")))
print ("Enter the elements of array 2")
for i in range(d):
    b = append(b, int(input("Enter the element ")))

c = add(a, b)

print(a)
print(b)
print(c)