Using numpy.argwhere makes code easier to read

This commit is contained in:
kpk09
2015-01-25 00:22:11 +00:00
parent c5ac705123
commit 305f5fd225
+1 -1
View File
@@ -146,7 +146,7 @@ def peak_local_max(image, min_distance=10, threshold_abs=0, threshold_rel=0.1,
peak_threshold = max(np.max(image.ravel()) * threshold_rel, threshold_abs)
# get coordinates of peaks
coordinates = np.transpose((image > peak_threshold).nonzero())
coordinates = np.argwhere(image > peak_threshold)
if coordinates.shape[0] > num_peaks:
intensities = image.flat[np.ravel_multi_index(coordinates.transpose(),image.shape)]