【发布时间】:2018-06-16 12:49:17
【问题描述】:
arr = np.array(range(12))
arr = arr.reshape((4,3))
keys = np.array([1,0,1,2])
arr[np.arange(4), np.array([1,0,1,2])]
这打印:
array([ 1, 3, 7, 11]).
我知道这是根据索引 (1,1) (0,0) (1,1) (2,2) 进行排序
但是怎么做呢?我不是指任何变量。我正在调用函数 np.array,现在我指的是 arr 的索引。为什么不创建新数组?
【问题讨论】:
标签: python arrays python-3.x numpy jupyter-notebook