numpy入门——数组切片

数组切片

• 选取第二行第一列的数字

# 导入numpy包 以np的形式import numpy as np# 使用arange的方式创建一个二维数组a = np.arange(1,13).reshape((3,4))print(a)# 选取第二行第一列的数字print(a[2,1])

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

• 选取第二行的数据

print(a[2,:])

array([ 9, 10, 11, 12])

选取第二行与第二三列的数字

print(a[2,2:4])

array([11, 12])

-迭代遍历每一行的数

for row in a:    print(row)

[1 2 3 4][5 6 7 8][ 9 10 11 12]

• 迭代遍历每一列数据

for column in a.T:    print(column)

[1 5 9][ 2  6 10][ 3  7 11][ 4  8 12]

• 迭代遍历数组的每一项

for item in a.flat:    print(item)

123456789101112

• 将数组组合成一维数组

print(a.flatten())

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