我有一个二维的numpy数组
对于每一列，我想在第N行中加1，其中N是该列第0行中的值。
我怎样才能做到这一点？
A=np.zeros(100)
A=np.reshape(A,[20,5])
A[0]=[5,2,4,1,3]
I want to add 1 to A[5,0], A[2,1], A[4,2], A[1,3] and A[3,4]
- 积分
0 - 话题
0 - 评论
3153 - 注册排名
1433
我有一个二维的numpy数组
对于每一列，我想在第N行中加1，其中N是该列第0行中的值。
我怎样才能做到这一点？
A=np.zeros(100)
A=np.reshape(A,[20,5])
A[0]=[5,2,4,1,3]
I want to add 1 to A[5,0], A[2,1], A[4,2], A[1,3] and A[3,4]
Following the following logic:
add 1 to A[5,0], A[2,1], A[4,2], A[1,3] and A[3,4], you can do so using advanced indexing: