calculo matriz

Esta é uma pré-visualização de arquivo. Entre para ver o arquivo original

def calculo_matriz():
 matriz_1=[]
 matriz_2=[]
 matriz_3=[]
 print("***** CÁLCULO DE MATRIZES *****")
 print("")
 print("MENU PRINCIPAL:")
 print("[1] SOMAR MATRIZ")
 print("[2] MULTIPLICAR MATRIZ")
 print("[0] SAIR")
 menu=input("OPÇÃO:")
 print("")
 
 while menu!=0:
 if menu==1:
 print("***** SOMA DE MATRIZES *****")
 print("")
 print("MATRIZ 1")
 lin_1=input("DIGITE QTDE DE LINHA:")
 col_1=input("DIGITE QTDE DE COLUNA:")
 
 for i in range(lin_1):
 linha_1=[]
 for j in range(col_1):
 linha_1.append(0)
 matriz_1.append(linha_1)
 for i in range(lin_1):
 for j in range(col_1):
 matriz_1[i][j]=input("DIGITE A MATRIZ 1(%d,%d):"%(i,j))
 print("")
 print("MATRIZ 2")
 lin_2=input("DIGITE QTDE DE LINHA:")
 col_2=input("DIGITE QTDE DE COLUNA:")
 for i in range(lin_2):
 linha_2=[]
 for j in range(col_2):
 linha_2.append(0)
 matriz_2.append(linha_2)
 for i in range(lin_2):
 for j in range(col_2):
 matriz_2[i][j]=input("DIGITE A MATRIZ 2(%d,%d):"%(i,j))
 if lin_1==lin_2 and col_1==col_2:
 #MATRIZ 3(SOMA DA MATRIZ_1 COM A MATRIZ_2)
 for i in range(lin_1):
 linha_3=[]
 for j in range(col_1):
 linha_3.append(0)
 matriz_3.append(linha_3)
 for i in range(lin_1):
 for j in range(col_1):
 matriz_3[i][j]=matriz_1[i][j]+matriz_2[i][j]
 
 print("")
 print("MATRIZ-1")
 for i in range(lin_1):
 for j in range(col_1):
 print matriz_1[i][j],
 print("")
 print("")
 print("MATRIZ-2")
 for i in range(lin_2):
 for j in range(col_2):
 print matriz_2[i][j],
 print("")
 print("")
 print("MATRIZ-3")
 print("(soma das matrizes 1 e 2)")
 for i in range(lin_1):
 for j in range(col_1):
 print matriz_3[i][j],
 print("")
 
 else:
 print("")
 print("AS MATRIZES NÃO SÃO IGUAIS")
 if menu==2:
 print("***** MULTIPLICAÇÃO DE MATRIZES *****")
 print("")
 print("MATRIZ 1")
 lin_1=input("DIGITE QTDE DE LINHA:")
 col_1=input("DIGITE QTDE DE COLUNA:")
 
 for i in range(lin_1):
 linha_1=[]
 for j in range(col_1):
 linha_1.append(0)
 matriz_1.append(linha_1)
 for i in range(lin_1):
 for j in range(col_1):
 matriz_1[i][j]=input("DIGITE A MATRIZ 1(%d,%d):"%(i,j))
 print("")
 print("MATRIZ 2")
 lin_2=input("DIGITE QTDE DE LINHA:")
 col_2=input("DIGITE QTDE DE COLUNA:")
 for i in range(lin_2):
 linha_2=[]
 for j in range(col_2):
 linha_2.append(0)
 matriz_2.append(linha_2)
 for i in range(lin_2):
 for j in range(col_2):
 matriz_2[i][j]=input("DIGITE A MATRIZ 2(%d,%d):"%(i,j))
 if lin_1==lin_2 and col_1==col_2:
 #MATRIZ 3(MULTIPLICAÇÃO DA MATRIZ_1 COM A MATRIZ_2)
 for i in range(lin_1):
 linha_3=[]
 for j in range(col_1):
 linha_3.append(0)
 matriz_3.append(linha_3)
 for i in range(lin_1):
 for j in range(col_1):
 matriz_3[i][j]=matriz_1[i][j]*matriz_2[i][j]
 
 print("")
 print("MATRIZ-1")
 for i in range(lin_1):
 for j in range(col_1):
 print matriz_1[i][j],
 print("")
 print""
 print("MATRIZ-2")
 for i in range(lin_2):
 for j in range(col_2):
 print matriz_2[i][j],
 print("")
 print("")
 print("MATRIZ-3")
 print("(multiplicação das matrizes 1 e 2)")
 for i in range(lin_1):
 for j in range(col_1):
 print matriz_3[i][j],
 print""
 
 else:
 print("")
 print("AS MATRIZES NÃO SÃO IGUAIS")
 print("***** CÁLCULO DE MATRIZES *****")
 print("")
 print("MENU PRINCIPAL:")
 print("[1] SOMA MATRIZ")
 print("[2] MULTIPLICAÇÃO MATRIZ")
 print("[0] SAIR")
 menu=input("OPÇÃO:")
 print("")
calculo_matriz()