python3-Numpy的索引

前言

numpy的索引,其实就是学习数组或矩阵的索引,类似于之前学过的数组a[2]

一维数组的索引

1
2
3
4
5
6
7
a = np.arange(3, 15)
print(a)
print(a[3]) # 输出一维数组中的第四个元素 :6

结果:
[ 3 4 5 6 7 8 9 10 11 12 13 14]
6

矩阵的索引

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
b = np.arange(3, 15).reshape((3, 4))
print(b)
print(b[2, 1]) # 或者b[2][1] :输出第3行第二列的元素
print(b[2, :]) # 输出第3行所有的元素
print(b[:, 1]) # 输出第2列所有的元素
print(b[1, 1:2]) # 输出第2行的第2列到第3列的值:8
print(b[1, 1:3]) # 输出第2行的第2列到第4列的值:8,9

#结果:
[[ 3 4 5 6]
[ 7 8 9 10]
[11 12 13 14]]
12
[11 12 13 14]
[ 4 8 12]
[8]
[8 9]

矩阵的迭代

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
for row in b:
print(row)
'''
等价于迭代转置矩阵的列:
for column in b.T:
print(column)
'''

for column in b:
print(column)

# 遍历整个矩阵
print(b.flatten())
for item in b.flat:
print(item)


#结果:
[3 4 5 6]
[ 7 8 9 10]
[11 12 13 14]

[3 4 5 6]
[ 7 8 9 10]
[11 12 13 14]

[ 3 4 5 6 7 8 9 10 11 12 13 14]

3
4
5
6
7
8
9
10
11
12
13
14
-------------The End-------------
谢谢大锅请我喝杯阔乐~