Как сложить два массива в numpy?

Пользователь

от velma , в категории: Python , 2 года назад

Как сложить два массива в numpy?

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

2 ответа

Пользователь

от fred , 2 года назад

@velma 

В Numpy есть несколько способов сложить два массива. Один из самых простых способов - это использовать функцию numpy.concatenate. Например, чтобы сложить два массива a и b, можно написать следующий код:

1
2
3
4
5
6
7
import numpy as np

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
c = np.concatenate([a, b])

print(c)  # выведет [1 2 3 4 5 6]


Функция concatenate может сложить несколько массивов сразу, передав их в виде списка. Например, чтобы сложить три массива a, b и c, можно написать следующий код:

1
2
3
4
5
6
7
8
import numpy as np

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
c = np.array([7, 8, 9])
d = np.concatenate([a, b, c])

print(d)  # выведет [1 2 3 4 5 6 7 8 9]


Если вы хотите сложить два массива по определенной оси, то можете использовать функцию numpy.concatenate с параметром axis. Например, чтобы сложить два массива a и b по вертикали (т.е. столбцы), можно написать след

Пользователь

от jaren , год назад

@velma 

ующий код:


1 2 3 4 5 6 7


import numpy as np


a = np.array([[1, 2, 3], [4, 5, 6]]) b = np.array([[7, 8, 9], [10, 11, 12]]) c = np.concatenate([a, b], axis=0)


print(c)

выведет

[[ 1 2 3]

[ 4 5 6]

[ 7 8 9]

[10 11 12]]

В данном примере массивы a и b имеют форму (2, 3) и их значения складываются по оси 0 (по вертикали), что приводит к форме (4, 3).