Python 使用 np.reshape 按特定顺序重塑数组

Question

我有一个数组 (a)，它的形状是 (1800,144)，其中 a[0:900,:] 都是实数，数组的后半部分 a[900:1800,:] 都是零。我想将数组的后半部分水平放置在前半部分的旁边，然后将它们推到一起，这样新的数组形状 (a) 将为 (900,288) 并且数组 a 将如下所示：

[[1,2,3,......,0,0,0],
 [1,2,3,......,0,0,0],
 ...
]

如果说得通。

当我尝试使用 np.reshape(a,(900,288)) 时，它并不完全符合我的要求。它使数组从 a[0:450,:] 开始全部为实数，从 a[450:900,:] 开始全部为零。我希望所有的零都被附加到第二个维度上，以便 a[0:900,0:144] 是所有实数并且 a[0:900,144:288] 都是零。

有没有简单的方法来做到这一点？

Answer 1

您可以使用 numpy.hstack() 连接两个数组：

import numpy as np
np.hstack([a[0:900,], a[900:1800,]])

如果你想将数组拆分成两个以上的子数组，你可以结合使用np.split和np.hstack，正如@HM14评论的那样：

np.hstack(np.split(a, n))  # assuming len(a) % n == 0 here

Answer 2

抱歉，这对于评论来说太大了，所以我会 post 在这里。如果你有一个长数组并且你需要拆分它并重新assemble它，还有其他方法可以完成这个。此示例说明如何 assemble 将相同大小的数字序列放入单个数组中。

a = np.arange(100)
>>> b = np.split(a,10)
>>> c = np.c_[b]
>>> c
array([[ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14, 15, 16, 17, 18, 19],
       [20, 21, 22, 23, 24, 25, 26, 27, 28, 29],
       [30, 31, 32, 33, 34, 35, 36, 37, 38, 39],
       [40, 41, 42, 43, 44, 45, 46, 47, 48, 49],
       [50, 51, 52, 53, 54, 55, 56, 57, 58, 59],
       [60, 61, 62, 63, 64, 65, 66, 67, 68, 69],
       [70, 71, 72, 73, 74, 75, 76, 77, 78, 79],
       [80, 81, 82, 83, 84, 85, 86, 87, 88, 89],
       [90, 91, 92, 93, 94, 95, 96, 97, 98, 99]])

因此您可以轻松拆分序列并轻松重新assemble。如果需要，您可以重新排序堆叠顺序。也许按此顺序更容易显示。

d = np.r_[b[5:],b[:5]].ravel()
>>> d
array([50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67,
       68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85,
       86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99,  0,  1,  2,  3,
        4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21,
       22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39,
       40, 41, 42, 43, 44, 45, 46, 47, 48, 49])

这个例子只是简单地展示了你可以把最后五个拆分序列放到堆的前面。很快就会发现，如果您有一系列值，即使长度不等，也可以将它们放在一个列表中，然后使用 np.c_ 和 np.r_ 重新 assemble 它们便利函数（np.c_ 通常期望大小相等的数组）。

所以也许不是针对您的具体情况的解决方案，而是关于如何以各种方式重新assemble 样本的一些建议。

Python 使用 np.reshape 按特定顺序重塑数组

Python reshaping array in a certain order using np.reshape

python

arrays

numpy

reshape