zebra/pyzebra
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update fit2.py

Browse Source 
Added user input to no peak case
This commit is contained in:
JakHolzer 2020-10-07 10:29:04 +02:00 committed by GitHub
parent 01ac35f93d
commit 7a50f2c8d5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pyzebra/fit2.py
View File

@ -65,11 +65,11 @@ def fitccl(
# Case for no peak, gaussian in centre, sigma as 20% of range # Case for no peak, gaussian in centre, sigma as 20% of range
print("No peak") print("No peak")
peak_index = find_nearest(x, np.mean(x)) peak_index = find_nearest(x, np.mean(x))
guess[0] = x[int(peak_index)] guess[0] = x[int(peak_index)] if guess[0] is None else guess[0]
guess[1] = (x[-1] - x[0]) / 5 guess[1] = (x[-1] - x[0])/5 if guess[1] is None else guess[1]
guess[2] = 10 guess[2] = 50 if guess[2] is None else guess[2]
guess[3] = 0 guess[3] = 0 if guess[3] is None else guess[3]
guess[4] = np.mean(y) guess[4] = np.mean(y) if guess[4] is None else guess[4]
constraints_min[2] = 0 constraints_min[2] = 0
elif len(meas["peak_indexes"]) == 1: elif len(meas["peak_indexes"]) == 1: