如何使numpy数组的列表都具有相同的形状？
如何使numpy数组的列表都具有相同的形状？
提问于 2013-05-30 10:36:25
所以我在列表中有一堆二维Numpy数组，我想确保它们都有相同的形状。我知道每个数组的第二个维度是相同的，但是第一个维度是不同的。
假设数组X的形状是(n，m)，数组Y的形状是(n+2，m)。我想将两行零添加到数组X中，这样X和Y都是(n+2，m)。
确保所有数组都具有相同形状的最具Python风格的方法是什么？假设我知道列表中所有数组的第一维的最大值是多少-称为N-并且，正如我前面提到的，我知道所有数组的第二维都是m。
谢谢大家！
回答 2
Stack Overflow用户
回答已采纳
发布于 2013-05-30 10:53:42
在一行代码中：
[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)])Stack Overflow用户
发布于 2013-05-30 11:11:11
最近，numpy.pad被引入了，所以还有：
>>> 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.]])页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/16827060
复制相关文章
相似问题
腾讯云开发者
