Как добавить два массива в 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)