將2D numpy數組垂直拆分為不均勻的子數組

Question

假設我有以下形狀(10, 5) numpy數組，我想將其拆分成兩個子數組：第一個包含前7行，第二個包含其余3行。 如果我這樣做：

x = np.arange(50).reshape(10, 5)
x1, y1 = np.vsplit(x, 2)

它將精確地分開一半。 我如何使其成為兩個子數組(7,5)和(3,5) ？

Answer 1

使用np.split() ：

In [4]: np.split(x, [7])
Out[4]: 
[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]]), array([[35, 36, 37, 38, 39],
        [40, 41, 42, 43, 44],
        [45, 46, 47, 48, 49]])]

Answer 2

我認為您應該使用奇特索引，而不像切片，奇特索引總是將數據復制到新數組中

n = 10; m = 5; i = 7
arr = np.arange(50).reshape(n, m)
arr7 = arr[np.ix_(range(i))]
arr3 = arr[np.ix_(range(i - n, 0, 1))]