将2D numpy数组垂直拆分为不均匀的子数组

时间:2018-06-02 08:39:00

标签: python python-3.x numpy

假设我有以下numpy形状(10, 5)数组,我希望将其分成两个子数组:第一个包含前7行,第二个包含剩余的3行。如果我这样做:

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

它将分成一半。如何将其设为两个子阵列(7,5)(3,5)

2 个答案:

答案 0 :(得分:2)

使用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]])]

答案 1 :(得分:0)

我认为你应该使用花哨的索引,不像切片,花哨的索引总是将数据复制到一个新的数组

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))]