好资源导航 » 文章资讯 » python 实现方阵的对角线遍历示例

python 实现方阵的对角线遍历示例

2023-08-05 06:05:03 97

任务描述

对一个方阵矩阵，实现平行于主对角线方向的对角线元素遍历。

从矩阵索引入手：

[[12345]
[678910]
[1112131415]
[1617181920]
[2122232425]]

上三角的索引遍历：

下三角的索引遍历：

代码

importnumpyasnp

A=np.arange(25)+1
A=np.mat(A.reshape([5,5]))
print(A)

"""
[[12345]
[678910]
[1112131415]
[1617181920]
[2122232425]]
"""

Num_element=A.shape[0]
c=int((Num_element-1)/2)
#print(c)


R=np.zeros_like(A)
#print(R)

forjinrange(Num_element):
print()
i=0
#print(i,j)
whilenp.max([i,j])
上述代码中对于每条对角线的所有元素执行相同的赋值操作。
考虑将其中重复的部分封装成函数：
defdiag_opreation(k,i,Num_element,R,A):
c=int((Num_element-1)/2)
whilenp.max([k,i])
则代码变为：
forjinrange(Num_element):
print()
i=0
#print(i,j)
R=diag_opreation(i,j,Num_element,R,A)

#print(R)

forkinrange(1,Num_element):
print()
i=0
#print(i,j)
R=diag_opreation(k,i,Num_element,R,A)

print(R)

输出结果为：
[[1311975]
[15131197]
[171513119]
[1917151311]
[2119171513]]
以上这篇python实现方阵的对角线遍历示例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。

返回顶部
514930285
czq8825@qq.com

python 实现方阵的对角线遍历示例

热门推荐

随机推荐