numpy.column_stack vectorized

Question

Я хотел был бы векторизовать некоторый код, но мне трудно применить некоторые функции к векторам переменных.

Например, у меня есть два постоянных вектора: a и b, а также вектор векторов x (матрица). Размеры элементов x такие же, как у a и b. Хочу сделать матрицы, образованные из столбцов: член x, a и b:

x = [[ 0.76662363 -0.0397725 0.64086377] 
[ 0.76198581 -0.04605764 0.6459538 ]] 

a = [ 0.2763932 0.85065081 -0.5527864 ] 

b = [-0.7236068 0.52573111 -0.5527864 ] 

Вывод должен быть вектор (или массив) двух матриц 3x3. Я пытаюсь выполнить следующий код:

a = np.column_stack((x, a, b)) 

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

File "/software/python/2.7.8/lib/python2.7/site-packages/numpy/lib/shape_base.py", line 317, in column_stack 
    return _nx.concatenate(arrays, 1) 
ValueError: all the input array dimensions except for the concatenation axis must match exactly 

Любые идеи?

Answer 1

Идущий с обновленными требованиями:

>>> N = 2 # n cols of x.T 
>>> K = 3 # x and a and b 
>>> M = 3 # len(a) 
>>> outstack = np.empty((N, M, K)) 
>>> outstack[..., 0] = x 
>>> outstack[..., 1] = a[None, :] 
>>> outstack[..., 2] = b[None, :] 

outstack складывает две матрицы 3х3.

Трюк здесь предопределяет выходной стек. Вывод служит ссылкой на радиовещание, поэтому a и b могут использоваться почти без изменений и будут правильно передаваться.