美文网首页
np.where()

np.where()

作者: 护国寺小学生 | 来源:发表于2019-01-31 19:01 被阅读0次

    numpy中能够返回符合某一条件的下标函数:np.where(),只不过np.where()并不接受list类型的参数。
    —— np.where()[0] 表示行的索引;
    —— np.where()[1] 则表示列的索引;

    ---------------------------------------------------------------------------

    np.where()用于三目运算:

    如果A%2==0成立,则执行A+1,否则执行A-1

    A
    array([1, 7, 4, 9, 2, 3, 6, 0, 8, 5])
    B = np.where(A%2 == 0, A+1, A-1) # 偶+1,奇-1
    B
    array([0, 6, 5, 8, 3, 2, 7, 1, 9, 4])

    相关文章

      网友评论

          本文标题:np.where()

          本文链接:https://www.haomeiwen.com/subject/mogzcqtx.html