numpy入门——数据分割

数据分割

• 等量分割

# 导入numpy包以np的形式import numpy as np# 创建一个3行4列的数组a = np.arange(1,13).reshape((3,4))print(a)# 等量分割print(np.split(a,4,axis=1))print(np.split(a,3,axis=0))

array([[ 1,  2,  3,  4],       [ 5,  6,  7,  8],       [ 9, 10, 11, 12]])[array([[1],        [5],        [9]]), array([[ 2],        [ 6],        [10]]), array([[ 3],        [ 7],        [11]]), array([[ 4],        [ 8],        [12]])][array([[1, 2, 3, 4]]), array([[5, 6, 7, 8]]), array([[ 9, 10, 11, 12]])]

• 非等量分割

print(np.array_split(a,3,axis=1))

[array([[ 1,  2],        [ 5,  6],        [ 9, 10]]), array([[ 3],        [ 7],        [11]]), array([[ 4],        [ 8],        [12]])]