假设我有一个numpy数组由A = np.zeros((4, 4, 4))定义。现在假设我想将元素A[0, 2, 3], A[1, 1, 2], A[2, 1, 0], A[3, 0, 3]设置为1。有没有一种方法可以在一行代码中完成？编辑：我试过A[[0, 2, 3], [1, 1, 2], [2, 1, 0], [3, 0, 3]] = 1，但没有工作。
A = np.zeros((4, 4, 4))
A[0, 2, 3], A[1, 1, 2], A[2, 1, 0], A[3, 0, 3]
A[[0, 2, 3], [1, 1, 2], [2, 1, 0], [3, 0, 3]] = 1
lymgl2op1#
当然，你可以简单地这样做：
A[0, 2, 3], A[1, 1, 2], A[2, 1, 0], A[3, 0, 3] = 1,1,1,1
字符串但我想你正在寻找先进的索引吗？
A[[0,1,2,3],[2,1,1,0],[3,2,0,3]] = 1
型
1条答案
按热度按时间lymgl2op1#
当然，你可以简单地这样做：
字符串
但我想你正在寻找先进的索引吗？
型