首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >numpy:如何获取数组反对角线?

numpy:如何获取数组反对角线?

提问于 2018-02-12 00:01:23
回答 2关注 0查看 1.1K

因此,在numpy数组中,有一个内置函数来获取对角线索引,但我似乎不知道如何从右上而不是左上角开始获得对角线。

这是从左上角开始的正常代码:

代码语言:txt
复制
>>> import numpy as np
>>> array = np.arange(25).reshape(5,5)
>>> diagonal = np.diag_indices(5)
>>> array
array([[ 0,  1,  2,  3,  4],
   [ 5,  6,  7,  8,  9],
   [10, 11, 12, 13, 14],
   [15, 16, 17, 18, 19],
   [20, 21, 22, 23, 24]])
>>> array[diagonal]
array([ 0,  6, 12, 18, 24])

但我想要它返回,

代码语言:txt
复制
array([ 4,  8, 12, 16, 20])
相关文章

相似问题

相关问答用户
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档