使用 einsum

问题描述

我只设法使用 Numpy einsum 提取了一条对角线。如何在 einsum 的帮助下获得 [6,37,68,99] 等其他对角线？

x =  np.arange(1,26 ).reshape(5,5)
y =  np.arange(26,51).reshape(5,5)
z =  np.arange(51,76).reshape(5,5)
t =  np.arange(76,101).reshape(5,5)
p =  np.arange(101,126).reshape(5,5)

a4 = np.array([x,y,z,t,p]

提取一条对角线：

>>>np.einsum('iii->i',a4)
>>>[  1  32  63  94 125]

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）