每天积累一点numpy

本博客记录工程上遇到的常用的却陌生的np用法

1.np.split（分割数组）

1.1 等分数组

In [2]: x = np.arange(9)

In [3]: x
Out[3]: array([0, 1, 2, 3, 4, 5, 6, 7, 8])

In [8]: np.split(x,3)
Out[8]: [array([0, 1, 2]), array([3, 4, 5]), array([6, 7, 8])]

In [9]: np.split(x,9)
Out[9]:
[array([0]),
 array([1]),
 array([2]),
 array([3]),
 array([4]),
 array([5]),
 array([6]),
 array([7]),
 array([8])]

1.2 按照元素的位置分割数组

In [5]: np.split(x,[2,5,6])
Out[5]: [array([0, 1]), array([2, 3, 4]), array([5]), array([6, 7, 8])]

In [6]: np.split(x,[1,3,5,7])
Out[6]: [array([0]), array([1, 2]), array([3, 4]), array([5, 6]), array([7, 8])]