# -*- coding: utf-8 -*-
####ID e3-2-0-compute

#### Άσκηση: Γράψτε μια συνάρτηση σε python:
####    def compute(A, B)
#### η οποία παίρνει ως παραμέτρους δύο numpy arrays ίδιου σχήματος n x n (δύο τετράγωνους πίνακες δηλ. ίδιου μεγέθους)
#### και επιστρέφει τον πίνακα ABA (γινόμενο). Το n (η διάσταση του πίνακα) δεν είναι πάντα η ίδια (οι πίνακες δηλ. μπορεί να είναι
#### 2x2, 3x3, 4x4 κλπ).
####
#### Για παράδειγμα, αν
####
####  A = np.array([[1, 2, 3], [3, 4, 6.7], [5, 9.0, 5]]) 
####  B = np.array([[1, 0, 1], [0, 0, 1], [1, -1, -1]])
####
#### τότε το αποτέλεσμα είναι
####
####  array([[ -5.  ,  -4.  ,  -8.1 ],
####         [ -8.9 ,  -4.7 , -14.29],
####         [ 40.  ,  81.  ,  41.5 ]])
####
#### Γράφετε τον κώδικά σας αποκλειστικά ανάμεσα στις γραμμές START και STOP παρακάτω.
#### Δεν πρέπει να πειράξετε καθόλου τις υπόλοιπες γραμμές
####
#### Για να τρέξετε το πρόγραμμά σας (αποθηκευμένο στο αρχείο user.py) δίνετε την εντολή
####
####    python user.py

import numpy as np

#### Απαιτήσεις:
####  Στο τέλος του προγράμματός σας θα πρέπει να είναι ορισμένη η συνάρτηση και να υπολογίζει αυτό που ζητάει
####  η άσκηση. Μη γράψετε εσείς κάποια κλήση στη συνάρτηση αλλά μόνο τον ορισμό της αρχίζοντας από τη γραμμή
####
####    def compute(A, B)
####
#### Μην αλλάξετε ό,τι είναι από τη γραμμή START και πάνω
####
####  ΔΕΝ ΕΠΙΤΡΕΠΟΝΤΑΙ εντολές print ή input στο πρόγραμμά σας
####
####START Από δω και κάτω γράφετε το πρόγραμμά σας.----------------------------------------------------------------

#### Για να δοκιμάζετε το πρόγραμμά σας με διαφορετικούς πίνακες μπορείτε να τροποιείτε τις επόμενες δύο γραμμές
AA = np.array([[1, 2, 3], [3, 4, 6.7], [5, 9.0, 5]])
BB = np.array([[1, 0, 1], [0, 0, 1], [1, -1, -1]])

def compute(A, B):

####STOP Μην αλλάξετε ό,τι είναι από τη γραμμή αυτή και κάτω--------------------------------------------------------
print "Result: "
print compute(AA, BB)