01. Training For New Users » 1_02 Simplify Tutorial of Python

# -*- coding: utf-8 -*-

"""

Created on Mon Mar 27 12:50:23 2017

Gamma function and dJ/dp

University of Sao Paulo

@author: Gabriel

"""

def Gamma(A,B,C):

    from numpy import *

    gamma = zeros((A.shape[2],A.shape[2]))

    dJdp = 0

    line = 0

    for i in A:

        gamma += dot(i.T[:,1:],i[1:,:])

        dJdp += dot(i.T[:,1:],array([(B-C)[line,1:]]).T)

        line += 1

    return gamma,dJdp