python3-Numpy的索引

前言

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

一维数组的索引

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

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

矩阵的索引

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]

矩阵的迭代

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