Python - Разбить матрицу numpy на положительные и отрицательные компоненты

Question

У меня есть numpy массивы формы (600,600,3), где значения - [-1.0, 1.0]. Я хотел бы расширить массив до (600,600,6), где исходные значения разделяются на суммы выше и ниже 0. Некоторые примеры (1,1,3) массивов, где th функция foo() делает трюк:

>>> a = [-0.5, 0.2, 0.9] 
>>> foo(a) 
[0.0, 0.5, 0.2, 0.0, 0.9, 0.0] # [positive component, negative component, ...] 
>>> b = [1.0, 0.0, -0.3] # notice the behavior of 0.0 
>>> foo(b) 
[1.0, 0.0, 0.0, 0.0, 0.0, 0.3] 

Answer 1

Использование нарезка для назначения мин/макс к различным частям выходного массива

In [33]: a = np.around(np.random.random((2,2,3))-0.5, 1) 

In [34]: a 
Out[34]: 
array([[[-0.1, 0.3, 0.3], 
     [ 0.3, -0.2, -0.1]], 

     [[-0. , -0.2, 0.3], 
     [-0.1, -0. , 0.1]]]) 

In [35]: out = np.zeros((2,2,6)) 

In [36]: out[:,:,::2] = np.maximum(a, 0) 

In [37]: out[:,:,1::2] = np.maximum(-a, 0) 

In [38]: out 
Out[38]: 
array([[[ 0. , 0.1, 0.3, 0. , 0.3, 0. ], 
     [ 0.3, 0. , 0. , 0.2, 0. , 0.1]], 

     [[-0. , 0. , 0. , 0.2, 0.3, 0. ], 
     [ 0. , 0.1, -0. , 0. , 0.1, 0. ]]])