我有一个看起来像这样的数组:
np.array([[0 , 5, 1], [0, 0, 3], [1, 7, 0]])
我想检查每个元素是否非零,如果它非零,则用一个数字替换它,以跟踪它检查了多少元素。也就是说,我希望最终产品看起来像
np.array([[0, 2, 3], [0, 0, 6], [7, 8, 0]])
其中第一行读取[0, 2, 3]
是因为第二个元素被第二个检查,通过了测试,然后被替换(等等)。谁能想到任何解决方案?我想numpy的索引在这里会非常有用。谢谢!
有只小跳蛙
泛舟湖上清波郎朗
相关分类