Python_numpy_索引

'''布尔索引'''
import numpy as np
array=np.arange(10)
#一维数组 案例1

# array_=array>5
# print(array_)#产生布尔类型
# print(array[array_])

#例2：把一维数组进行01化

'''array_1=array>5
array_0=array<=5
array[array_1]=1
array[array_0]=0
array[array<=5]=0
或者使用下面2句话
'''
array[array>5]=1#注意两句话的先后顺序
print(array)

#例3进行自增量的操作，给大于5的加上520

array[array>5]+=520
print(array)

 

#二维数组案例1:
tarray=np.arange(1,21).reshape(4,5)
print(tarray)
# print(tarray[tarray>10])
#例2:把第三列 大于5的行筛选出来并重新赋值为520
# tarray1=tarray[:,3]>5 #找出第三列所有大于5的
# tarray[:,3][tarray1]=520#tarray[:,3]找到第三列,然后和上一步结合，找到第三列大于5的，然后变成520
# print(tarray)
#例3：找出偶数或小于7的数
tarray=np.arange(10)
tarray3=(tarray%2==0)|(tarray<7)
print(tarray3)
print(tarray[tarray3])

'''神奇索引：使用整数数组进行数据索引
一维数组的神奇索引:a[[1,2,3]] 代表对应的数组值
二维数组的神奇索引:a[[1,2,3]] 代表对应的行
a[[2,3],[4,5]] 代表对应的数组中的值
'''
 

import numpy as np
a=np.arange(32).reshape(4,8)
print(a)
print(a[[0,2,3]])#选出二维数组a的0、2、3行
print(a[2:4,3])
b=np.arange(10)
print(b[[3,4,5]])

#例1获取数组中最大的前N个数字
import numpy as np
a=np.random.randint(1,100,10)
b=a.argsort()[-3:] #对a进行排序（升序）并获取排好序的下标，再取后三个较大值的下标
print(a[b])