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)

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

• 将数组组合成一维数组

print(a.flatten())

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