numpy索引(三)

1.一维索引

a = np.arange(3,15)
print(a)
print(a[3])#取下标为3的索引
result:
[ 3  4  5  6  7  8  9 10 11 12 13 14]
6

2.二维索引

b = a.reshape(3,4)
print(b)
print(b[1])#取第一行
print(b[0,1])#取第0行,第1列
print(b)
print(b[1,:])#取b的第一行,所有列
print(b[:,1])#取b的所有行,第一列
print(b[1,0:1])#取第一行,0到1列的元素
result:
[[ 3  4  5  6]
 [ 7  8  9 10]
 [11 12 13 14]]
[ 7  8  9 10]
4
4
[[ 3  4  5  6]
 [ 7  8  9 10]
 [11 12 13 14]]
[ 7  8  9 10]
[ 4  8 12]
[7]

3.迭代打印

for row in b:
    print(row)#将行迭代索引打印

for column in b.T:
    print(column)#无列的打印,但可以通过转置,打印出列

print(b.flatten())#将矩阵压缩为一维矩阵

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

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

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