扫码加入数据分析学习群

# -*- coding:utf-8 -*-
# author:
import numpy
'''ndarray的布尔型索引'''
x = numpy.array([3,2,3,1,3,0])
# 布尔型数组的长度必须跟被索引的轴长度一致
y = numpy.array([True,False,True,False,True,False]) 
print(x[y] )# [3,3,3] 
print(x[y==False]) # [2,1,0]
print(x>=3) # [ True False  True False  True  False]
print(x[~(x>=3)]) # [2,1,0]
print((x==2)|(x==1) )# [False  True False  True False False]
print(x[(x==2)|(x==1)] )# [2 1]
x[(x==2)|(x==1)] = 0
print(x )# [3 0 3 0 3 0]