конкатенация два многомерные массивы в NumPy

Question

У меня есть два массива A и B,

>> np.shape(A) 
>> (7, 6, 2) 
>> np.shape(B) 
>> (6,2) 

Теперь я хочу, чтобы объединить два массива таким образом, что A распространяется на (8,6,2) с A[8] = B

Я попытался np.concatenate()

>> np.concatenate((A,B),axis = 0) 
--------------------------------------------------------------------------- 
ValueError        Traceback (most recent call last) 
<ipython-input-40-d614e94cfc50> in <module>() 
----> 1 np.concatenate((A,B),axis = 0) 

ValueError: all the input arrays must have same number of dimensions 

и np.vstack()

>> np.vstack((A,B)) 
--------------------------------------------------------------------------- 
ValueError        Traceback (most recent call last) 
<ipython-input-41-7c091695f277> in <module>() 
----> 1 np.vstack((A,B)) 
//anaconda/lib/python2.7/site-packages/numpy/core/shape_base.pyc in vstack(tup) 
    228 
    229  """ 
--> 230  return _nx.concatenate([atleast_2d(_m) for _m in tup], 0) 
    231 
    232 def hstack(tup): 

ValueError: all the input arrays must have same number of dimensions 

Answer 1

Вероятно, самый простой способ заключается в использовании NumPy newaxis так:

import numpy as np 

A = np.zeros((7, 6, 2)) 
B = np.zeros((6,2)) 
C = np.concatenate((A,B[np.newaxis,:,:]),axis=0) 
print(A.shape,B.shape,C.shape) 

, что приводит к следующим образом:

(7, 6, 2) (6, 2) (8, 6, 2) 

Как @sascha упоминалось вы можете использовать vstack (также см hstack, dstack) для выполнения операций прямой конкатенации с неявной осью (соответственно axis = 0, axis = 1, axis =2):

D = np.vstack((A,B[np.newaxis,:,:])) 
print(D.shape) 

, результат:

(8, 6, 2)