Numpy入门(八)：np.piecewise()用法

点击跳转 《Numpy入门系列目录》

---

numpy.piecewise(x, condlist, funclist, *args, **kw)

x： 表示要进行操作的对象

condlist： 表示要满足的条件列表，可以是多个条件构成的列表

funclist： 执行的操作列表，参数二与参数三是对应的，当参数二为true的时候，则执行相对应的操作函数

功能：根据相关的条件，进行筛选，然后对满足不同条件的元素进行相关的操作，这个操作可以来源与函数、lambda表达式等，并得到新的结果，返回一个array对象，和原始操作对象x具有完全相同的维度和形状

例子

import numpy as np x = np.arange(0, 10) print(x) # [0 1 2 3 4 5 6 7 8 9] print(np.piecewise(x, [x < 4, x >= 6], [-1, 1])) # [-1 -1 -1 -1 0 0 1 1 1 1] # 将元素中小于4的用-1替换掉，大于等于6的用1替换掉，其余的默认以0填充