02 numpy 索引与切片

切片

一维数组的切片

import numpy as np
a=np.arange(10)
print(a)
print("--------------")
print(a[2:7:2])

二维数组的切片

import numpy as np
a=np.arange(10).reshape(2,5)
print(a)
print("--------------")
print(a[0,3])  #获取0行3列的数字
print("--------------")
print(a[1]) #获取第1行
print("--------------")
print(a[-1]) #获取最后一行
print("--------------")
print(a[0:1,1:3]) #获取0行,1-2列

bool索引

import numpy as np
a=np.arange(10)
#筛选大于5的数字
b=a>5
print(b)
print(a[b])

数组大于5的变成1，小于5的变成0

注意：先写小于，再写大于

import numpy as np
a=np.arange(10)
print(a)
a[a<=5]=0
a[a> 5]=1
print(a)

数组大于5的加520，小于5的变成0

import numpy as np
a=np.arange(10)
print(a)
a[a<=5]=0
a[a> 5]+=520
print(a)

二维数组

import numpy as np
a=np.arange(1,21).reshape(4,5)
print(a)
print('------------------')
a[a<=5]=0
a[a> 5]+=520
print(a)

大于7的偶数

import numpy as np
a=np.arange(0,10)
print(a)
print('------------------')
#大于7的偶数
b=(a>7 )&( a%2==0)
print(a[b])

神奇索引

获取 1，2，3 行

import numpy as np
a=np.arange(36).reshape(9,4)
print(a)
print('------------------')
print(a[[3,4,5]]) # 获取 1，2，3 行 

获取 1，2，3 行 的1，2，3列

import numpy as np
a=np.arange(36).reshape(4,9)
print(a)
print('------------------')
print(a[[1,2,3]]) # 获取1,2,3 行
print('------------------')
print(a[[1,2,3],[1,2,3]])

取所有行的第1，2列

import numpy as np
a=np.arange(36).reshape(4,9)
print(a)
print('------------------')
print(a[:,[1,2]]) 

特殊

import numpy as np
a=np.arange(10)
print(a)
print("-"*30)
index=np.array([[0,2],[1,3]])
print(a[index])

实例

数组的转置transpose 方法

import numpy as np
a=np.arange(10).reshape(2,5)
print(a)
print("-"*30)
print(a.transpose())