SPMUtilities/AFM_thumbs
2
3
Fork 0
Code Issues Pull Requests Releases Wiki Activity

If all data pixel are on NaN, then assign a value of 0.0 for each pixel.

Browse Source
This commit is contained in:
Clemens Barth
2023-04-26 15:36:08 +02:00
parent ab9fd59148
commit a3492c10f3
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
AFM_thumbs/Linalg.py
View File

@@ -119,7 +119,7 @@ def linalg_line_fit(size_x, size_y, data):
xy = sum(line*x_vec) xy = sum(line*x_vec)
B = scipy.mat([[xy],[y]]) B = scipy.mat([[xy],[y]])
c,resid,rank,sigma = linalg.lstsq(M,B) c, resid, rank, sigma = linalg.lstsq(M, B)
plane.append([i * c[0][0] + c[1][0] for i in range(size_x)]) plane.append([i * c[0][0] + c[1][0] for i in range(size_x)])
@@ -198,7 +198,12 @@ def check_image_properties(data, data_file):
list_nans = np.where(np.isnan(data))[0] list_nans = np.where(np.isnan(data))[0]
data_clean = np.delete(data, list_nans) data_clean = np.delete(data, list_nans)
data_mean = np.mean(data_clean)
# If all pixels are on NaN, then put the mean value onto 0.0.
if len(list_nans) == len(data):
data_mean = 0.0
else:
data_mean = np.mean(data_clean)
for nan in list_nans: for nan in list_nans:
data[nan] = data_mean data[nan] = data_mean