from pylab import *
from numpy import *  #je dobre zvyknutis si aj na tento import
#uvedene dva importy v neinteraktivne spustanych .py suboroch robia nieco podobne
# ako ked sa interaktivny ipython spustkik s option --pylab 

phi=pi/3.
dphi=phi/1000.

smallR=zeros([3,3])  #array 3x3 zo samych nul, pozor indexy prebiehaju mnozinu 0,1,2
smallR[0,0]=1
smallR[1,1]=1
smallR[2,2]=1
smallR[0,1]=-dphi
smallR[1,0]=dphi
print(smallR)  	#tlac pre konttrolu, co som urobil, ma to byt matica rotacie okolo osi
				#z o velmi maly uhol dphi, porovnajte s prednaskou

R=zeros([3,3])
R[0,0]=1
R[1,1]=1
R[2,2]=1
print(R)   #tlac pre kontrolu ma to byt jednotkova matica

#teraz vyrobime funkciu pre nasobenie matic
def matproduct(mat1,mat2):
	vysledok=zeros([3,3])
	for i in range(3):
		for j in range(3):
			for k in range(3):
				vysledok[i,j]=vysledok[i,j]+mat1[i,k]*mat2[j,k]  #!!!!!!!!!!!!!!!! v tomto riadku je chyba, opravte ju
	return vysledok


for i in range(1000):
	R=matproduct(smallR,R)
	
print(R)	
print("porovnajte s ocakavanou rotacnou maticou pre uhol pi/3")