18.numpy.where的使用

欢迎访问个人网络日志🌹🌹知行空间🌹🌹

numpy.where的使用

函数介绍

numpy.where(condition, [x, y, ])

• condition: 条件
• x:数组，shape同condition或支持广播
• y:数组，shape同condition或支持广播

根据condition从x或y中取数值，如果condition条件为True取x对应位置的元素，否则取y对应位置的元素。

当数组都是一维的时候，函数的功能等同于以下语句：

[xv if c else yv for c, xv, yv in