numpy学习index

一.一维二维的索引

import numpy as np
a=np.arange(3,15)#3-14,
print(a)
print(a[3])   #6，一维，第四个
b=np.array([[1,2,3],[4,5,6],[7,8,9]])
print(b)#3行3列的矩阵
print(b[1][1])  #5，第一行第一个
print(b[2,1])   #8，第2行第一个
print(b[2,:])   # [7 8 9]第2行的所有数字用冒号
print(b[:,1])   # [2,5,8]第1列的所有 

二.矩阵的遍历

import numpy as np
a=np.arange(3,15).reshape((3,4))#3-14,
print(a)
for row in a:
    print(row)
for column in np.transpose(a):
    print(column)

输出结果：
[[ 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 7 11]
[ 4 8 12]
[ 5 9 13]
[ 6 10 14]
三.矩阵的展开
flatten是一个展开性质的函数，将多维的矩阵进行展开成1行的数列。而flat是一个迭代器，本身是一个object属性

import numpy as np
a=np.arange(3,15).reshape((3,4))#3-14,
print(a)
print(a.flatten())
for item in a.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