如何將數組拆分為不相等的部分？

Question

我有這樣的數組

import numpy as np

x = np.array([1, 2, 3, 99, 99, 3, 2, 1])

我想把它分成三部分

x1 = array([1, 2, 3])
x2 = array([99, 99])
x3 = array([3, 2, 1])

做這個的最好方式是什么？

Answer 1

您可以使用np.split ：

x = np.array([1, 2, 3, 99, 99, 3, 2, 1])

x1, x2, x3 = np.split(x, [3, 5])

[3, 5]指定要分割的索引。

那產生

x1
array([1, 2, 3])

x2
array([99, 99])

x3
array([3, 2, 1])