当前位置 主页 > 服务器问题 > Linux/apache问题 >

    numpy.ndarray 实现对特定行或列取值

    栏目:Linux/apache问题 时间:2019-12-07 10:22

    如下所示:

    import numpy as np
     
    b = [[1,2,0],
    [4,5,0],
    [7,8,1],
    [4,0,1],
    [7,11,1]
      ]
    a=np.array([b]).reshape((5,3))
    print(a)
    c=[1,3,4]
    # print(a[c])
    d=np.nonzero(a[:, 2] == 0)
    print(d)
    print(a[d])
     
     
    

    以上这篇numpy.ndarray 实现对特定行或列取值就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持IIS7站长之家。