本文共 302 字,大约阅读时间需要 1 分钟。
使用ravel()
a = np.array(np.random.randint(0,20,(15,2)))b = np.array(np.random.randint(0,2,(15,1))).astype(np.float32)#若果使用a[b==1]就会报错,使用ravel()转化成以为数组a[b.ravel()==1]Out[38]: array([[ 6, 4], [19, 1], [ 4, 6], [ 1, 4], [17, 2], [ 1, 4], [ 4, 7], [ 9, 2]])
转载地址:http://xhben.baihongyu.com/