Python 如何列出所有形状相同的numpy数组?
所以我在一个列表中有一堆2d Numpy数组,我想确保它们都具有相同的形状。我知道每个数组的第二维是相同的,但是第一维是不同的 假设数组X的形状是(n,m),数组Y的形状是(n+2,m)。我想向数组X添加两行零,这样X和Y都是(n+2,m) 检查列表的最有效方法是什么,以确保所有数组具有相同的形状?假设我知道列表中所有数组的第一维的最大值是多少——称之为N——正如我提到的,我知道所有数组的第二维都是m 谢谢大家 在一行中:Python 如何列出所有形状相同的numpy数组?,python,numpy,Python,Numpy,所以我在一个列表中有一堆2d Numpy数组,我想确保它们都具有相同的形状。我知道每个数组的第二维是相同的,但是第一维是不同的 假设数组X的形状是(n,m),数组Y的形状是(n+2,m)。我想向数组X添加两行零,这样X和Y都是(n+2,m) 检查列表的最有效方法是什么,以确保所有数组具有相同的形状?假设我知道列表中所有数组的第一维的最大值是多少——称之为N——正如我提到的,我知道所有数组的第二维都是m 谢谢大家 在一行中: [np.r_[a, np.zeros((N - a.shape[0],
[np.r_[a, np.zeros((N - a.shape[0], m), dtype=a.dtype)] for a in your_arrays]
可能更容易阅读
for i,a in enumerate(your_arrays):
rows, cols = a.shape
if rows != N:
your_arrays[i] = np.vstack([a, np.zeros((N - rows, cols), dtype=a.dtype)])
相对而言,最近引入了,因此还有:
>>> X = np.ones((3,2))
>>> Y = np.ones((5,2))*2
>>> N = 5
>>> nX, nY = [np.pad(m, ((0,N-m.shape[0]),(0,0)), 'constant') for m in [X, Y]]
>>> nX
array([[ 1., 1.],
[ 1., 1.],
[ 1., 1.],
[ 0., 0.],
[ 0., 0.]])
>>> nY
array([[ 2., 2.],
[ 2., 2.],
[ 2., 2.],
[ 2., 2.],
[ 2., 2.]])
这很好,代码很好,紧凑,漂亮。谢谢,@wim!