Concatenate two numpy arrays in the 4th dimension

后端 未结 6 1444
一生所求
一生所求 2020-12-14 06:59

I have two numpy arrays with three dimensions (3 x 4 x 5) and I want to concatenate them so the result has four dimensions (3 x 4 x 5 x 2). In Matlab, this can be done with

6条回答
  •  遥遥无期
    2020-12-14 07:33

    It's not necessarily the most elegant, but I've used variations of

    c = rollaxis(array([a,b]), 0, 4)
    

    in the past.

提交回复
热议问题