np.where和np.bincount详解

来源:互联网 发布:linux 命令下载libgcj 编辑:程序博客网 时间:2024/05/29 09:41

原文地址:https://stackoverflow.com/questions/23359886/selecting-rows-in-numpy-ndarray-based-on-the-value-of-two-columns?rq=1

简单用法

a = np.array(xrange(5*3))
a.shape =(5,3)

对这样一个数组进行操作:
array([[0,1,2],[3,4,5],[6,7,8],[9,10,11],[12,13,14]])

  1. 寻找以9开头的某一行:
    a[np.where(a[:,0]==9)] —->#array([[ 9, 10, 11]])

  2. 寻找l列中某一值等于4的某一行:
    a[np.where(a[:,1]==4)] —->array([[3, 4, 5]])

  3. 寻找某一行:第一位值为0,第二位值为1:
    a[np.where((a[:,0]==0)*(a[:,1]==1))]#array([[0,1, 2]])

np.bincount详解

原文链接:
http://blog.csdn.net/xlinsist/article/details/51346523

原创粉丝点击