2つの配列の要素が交互に来るように結合

Question

前提・実現したいこと

pythonを学びたてのものです。初歩的な質問かもしれませんが
以下の2つの配列を3つ目のように結合させたいです。
a
[[264.39134912 268.09886715 291.33264676 335.57569516]
[335.57569516 335.57569516 364. 364. ]
[364. 324.45314109 304.43254377 303.19670443]
[303.19670443 303.19670443 264.39134912 264.39134912]]
b
[[ 32.47288503 155.37527115 196.17787419 258. ]
[258. 258. 225.11062907 225.11062907]
[225.11062907 183.07158351 138.80694143 30. ]
[ 30. 30. 32.47288503 32.47288503]]

[[264.39134912, 32.47288503,268.09886715,155.37527115,291.33264676,196.17787419,335.57569516,258]
：
[[303.19670443,30.,…264.39134912,32.47288503]]

発生している問題・エラーメッセージ

hstackなどを使用したところ
[[264.39134912 268.09886715 291.33264676 335.57569516]
[335.57569516 335.57569516 364. 364. ]
[364. 324.45314109 304.43254377 303.19670443]
[303.19670443 303.19670443 264.39134912 264.39134912]]
[[ 32.47288503 155.37527115 196.17787419 258. ]
[258. 258. 225.11062907 225.11062907]
[225.11062907 183.07158351 138.80694143 30. ]
[ 30. 30. 32.47288503 32.47288503]]
上記のように上下で結合するだけでした。

Answer 1

単純に代入でも対応できます。

import numpy as np

A = np.array([[1.,3.,5.,7., 9.],[11.,13.,15.,17.,19.]])
B = np.array([[2.,4.,6.,8.,10.],[12.,14.,16.,18.,20.]])

res = np.empty((A.shape[0], A.shape[1] + B.shape[1]),dtype=float)
res[:,0::2] = A
res[:,1::2] = B
print(res)
#[[ 1.  2.  3.  4.  5.  6.  7.  8.  9. 10.]
# [11. 12. 13. 14. 15. 16. 17. 18. 19. 20.]]

Comments

[bump_fuuu]

簡潔で分かりやすいご回答ありがとうございます！

Answer 2

numpy.insertを使うと以下のように書けます。

import numpy as np

a = np.array([[1,2,3,4],[5,6,7,8]])
b = np.array([[9,10,11,12],[13,14,15,16]])

i = range(a.shape[1])
c = np.insert(b,i,a[:,i],axis=1)
print(c)
"""
[[ 1  9  2 10  3 11  4 12]
 [ 5 13  6 14  7 15  8 16]]
"""

Comments

[bump_fuuu]

ご回答ありがとうございます！出来ました！内包表記などを使用した方がいいかなど思っていましたが簡単にできるんですね！

[8524ba23]

解決してよかったです。
なお、速度面では、あらかじめ領域を準備しておく＋スライス表現のmagichanさんの回答のほうが有利かと思います。