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