Use a consistent style of np function calls

2020-04-22 16:48:30 +02:00
parent 0a42a9ee8d
commit a287b78245
1 changed files with 1 additions and 1 deletions
--- a/pyzebra/xtal.py
+++ b/pyzebra/xtal.py
@ -374,7 +374,7 @@ def box_int(file,box):

 # omega fit
    om=dat["rot_angle"][fr0:frN]
-    cnts = dat["data"][fr0:frN,j0:jN,i0:iN].sum(axis=(1,2))
+    cnts = np.sum(dat["data"][fr0:frN,j0:jN,i0:iN], axis=(1,2))

    p0 = [1., 0., 1.]
    coeff, var_matrix = curve_fit(gauss, range(len(cnts)), cnts, p0=p0)