http://docs.scipy.org/doc/numpy/reference/arrays.indexing.html
numpy.newaxis
newaxis对象可用于所有切片 *** 作中,以创建长度为一的轴。:const:newaxis是’None’的别名,可以使用’None’代替它,结果相同。
http://docs.scipy.org/doc/numpy-1.10.1/reference/generated/numpy.expand_dims.html
演示部分代码
In [154]: labels=np.array([1,3,5])In [155]: labels[:,None]Out[155]: array([[1], [3], [5]])In [157]: np.arange(8)==labels[:,None]Out[157]: array([[False, True, False, False, False, False, False, False], [False, False, False, True, False, False, False, False], [False, False, False, False, False, True, False, False]], dtype=bool)In [158]: (np.arange(8)==labels[:,None]).astype(int)Out[158]: array([[0, 1, 0, 0, 0, 0, 0, 0], [0, 0, 0, 1, 0, 0, 0, 0], [0, 0, 0, 0, 0, 1, 0, 0]])
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)