假设你有一个3x3的数组x,如下所示:
>>> xarr
array([[0., 0., 0.],
[0., 0., 0.],
[1., 1., 1.]])
类似地,y数组如下:
>>> yarr
array([[1., 0., 0.],
[1., 0., 0.],
[1., 0., 0.]])
问题是有没有一种方法可以合并(或以任何向量化的方式使用)xarr和yarr,以获得如下结果数组:
>>> result
array([[1., 0., 0.],
[0., 1., 0.],
[0., 0., 1.]])
这个想法是在不使用循环的情况下获得结果数组,当然也没有numpy.diag或numpy.eye等。
1条答案
按热度按时间omjgkv6w1#
这是一种方法。