Numpy的索引

import numpy as np
A=np.arange(3,15)
print(A)
print(A[3])

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

#转成二维
B = np.arange(3,15).reshape((3,4))
print(B)

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

#查找第一行
print(B[0])

[3 4 5 6]

#查找第0行，第0列的数
print(B[0,0])
print(B[0][0])

3
3

#第二行的所有数
print(B[2,:])

[11 12 13 14]

#查找第1列的所有数
print(B[:,1])

[ 4 8 12]

#查找第1行，第1列到第3列的数
print(B[1,1:4])

[ 8 9 10]

#用for循环打印每一行
for row in B:
    print(row)

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

#不能直接打印出列，通过转置打印每一列
for column in B.T:
    print(column)

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

#用flatten返回一个折叠成一维的数组
print(B.flatten())
print(B.flat)

[ 3  4  5  6  7  8  9 10 11 12 13 14]
<numpy.flatiter object at 0x0000012C60CC60D0>