用“元素下标”替换数组中的元素
我有一个看起来像这样的数组：
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的索引在这里将非常有用。谢谢！
展开
收起
1
条回答
写回答
写回答
-
你可以做：
np.where(a == 0, a, np.arange(a.size).reshape(a.shape) + 1)
2019-09-27 09:53:59赞同 展开评论 打赏
版权声明：本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。
相关问答
