我有兩個 numpy 陣列,它們的大小(1,5,2)和(1, 1,2). 我想將其中兩個放在一個陣列中并用 (1, 6,2) 構建一個陣列,然后洗掉矩陣的第一行,最終陣列的大小應為(1,5,2).
import numpy as np
a = np.random.rand(1, 5, 2)
b = np.random.rand(1, 1, 2)
例如,
a = array([[[0.3324251 , 0.23363885],
[0.58601194, 0.56248613],
[0.38604242, 0.29020216],
[0.11876984, 0.6805301 ],
[0.67216552, 0.43272049]]])
b = array([[[0.25492274, 0.43373311]]])
輸出:
a = array([[[0.58601194, 0.56248613],
[0.38604242, 0.29020216],
[0.11876984, 0.6805301 ],
[0.67216552, 0.43272049],
[0.25492274, 0.43373311]]])
uj5u.com熱心網友回復:
您可以使用hstack,然后洗掉first row。
x = np.hstack((a, b))[:, 1:,:]
x = array([[[0.58601194, 0.56248613],
[0.38604242, 0.29020216],
[0.11876984, 0.6805301 ],
[0.67216552, 0.43272049],
[0.25492274, 0.43373311]]])
uj5u.com熱心網友回復:
使用堆疊:
np.hstack((a, b)) # h is for horizontal, if you want to put an array above another one, use np.vstack
轉載請註明出處,本文鏈接:https://www.uj5u.com/caozuo/441535.html
上一篇:將numpy2d陣列轉換為3d
