标签: python arrays numpy
我有一个n - 维数组arr,我需要在其中修改k条目,这些条目在形状为{{1}的单独数组pos中指定}。如果维度(k, n)已修复,则可以使用n轻松实现。如何将其推广到arr[pos[:, 0], pos[:, 1]] = 1维度?
n
arr
k
pos
(k, n)
arr[pos[:, 0], pos[:, 1]] = 1
特定案例n的工作示例是
n=3