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填充