import numpy as np def getKeff(pm,k,pbc,Nz): nx = k.shape[2] #Pasar k sin bordes de k=0 ny = k.shape[1] tz = 2*k[1,:,:]*k[0, :,:]/(k[0, :,:]+k[1,:,:]) q=((pm[0,:,:]-pm[1,:,:])*tz).sum() area=ny*nx l=Nz keff=q*l/(pbc*area) return keff,q