矩阵乘法 python 基础练习 vip

最新推荐文章于 2025-12-18 16:41:12 发布

原创最新推荐文章于 2025-12-18 16:41:12 发布 · 86 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#矩阵 #算法 #numpy #python #蓝桥杯

蓝桥杯专栏收录该内容

22 篇文章

订阅专栏

该代码段是用于计算一个N阶矩阵的M次幂的Python程序。它首先读取矩阵的阶数N和幂次M，然后接收矩阵的输入，通过多重循环进行矩阵乘法来计算结果，并按指定格式输出。

问题描述

　　给定一个N阶矩阵A，输出A的M次幂（M是非负整数）
　　例如：
　　A =
　　1 2
　　3 4
　　A的2次幂
　　7 10
　　15 22

输入格式

　　第一行是一个正整数N、M（1<=N<=30, 0<=M<=5），表示矩阵A的阶数和要求的幂数
　　接下来N行，每行N个绝对值不超过10的非负整数，描述矩阵A的值

输出格式

　　输出共N行，每行N个整数，表示A的M次幂所对应的矩阵。相邻的数之间用一个空格隔开

import copy
n=input()
n=n.split()
n=list(map(int,n))
l=[]
l_add=[0 for i in range(n[0])]#存储每行运算后的结果
for i in range(n[0]):
    m=input()
    m=m.split()
    m=list(map(int,m))
    l.append(m)
l_copy=copy.deepcopy(l)
'''幂次循环'''
for i1 in range(1,n[1]):
    '''行数循环'''
    for i2 in range(n[0]):
        '''列数循环'''
        for i3 in range(n[0]):
            s=0
            for i4 in range(n[0]):
                s+=l_copy[i2][i4]*l[i4][i3]
            l_add[i3]=s
            #print(l_add)
        '''将算好的行结果赋给要输出的矩阵'''
        for i5 in range(n[0]):
            l_copy[i2][i5]=l_add[i5]
            l_add[i5]=0
if n[1]==0:
    for i in range(n[0]):
        for j in range(n[0]):
            if i==j:
                print(1,end=" ")
            else:
                print(0,end=' ')
        print()
else:
    for i in range(n[0]):
        for j in range(n[0]):
            print(l_copy[i][j],end=" ")
        print()