numpy之where的用法

    技术2025-05-24  52

    1、numpy.where的返回结果

    numpy.where调用方式为numpy.where(condition,1,2)

    满足条件的位置上返回结果1,不满足的位置上返回结果2

     例如通过where()函数将a数组中负值设为0,正值不变

    如果没有指定返回结果,只有查找条件则返回满足条件的位置。返回的结果是一个元组(tuple),包含两个数组,第一个数组纪录的是行,第二个数组纪录的是列。

    可以使用zip函数将返回的位置组成一个个坐标对,方便调用。zip函数直接返回的是一个对象,可以用过for循环遍历出里面的元素,也可以使用list直接列出所有坐标对元素。

    2、numpy.where多条件查询

    与: numpy.where((con1)*(con2))或者用&

    或:numpy.where((con1)|(con2)) (重点:多条件查询时条件一定要用括号!一定要用括号!一定要用括号!)

    Processed: 0.008, SQL: 9