Skip to content

Commit e998c78

Browse files
committed
Switching to np.argmin
1 parent 126f03c commit e998c78

2 files changed

Lines changed: 1 addition & 16 deletions

File tree

src/ctapipe/image/statistics.py

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
from ..containers import ImageStatisticsContainer
77

88
__all__ = [
9-
"argmin",
109
"arg_n_largest",
1110
"arg_n_largest_gu",
1211
"n_largest",
@@ -140,15 +139,3 @@ def arg_n_largest(n, array):
140139
dummy = np.zeros(n, dtype=np.int64)
141140
idx = arg_n_largest_gu(dummy, array)
142141
return idx
143-
144-
145-
@njit
146-
def argmin(array):
147-
"""Returns the index of the minimum value of an array"""
148-
min_index = 0
149-
min_value = array[0]
150-
for i in range(1, len(array)):
151-
if array[i] < min_value:
152-
min_value = array[i]
153-
min_index = i
154-
return min_index

src/ctapipe/reco/telescope_event_handling.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,6 @@
77
import numpy as np
88
from numba import njit, uint64
99

10-
from ctapipe.image.statistics import argmin
11-
1210
__all__ = [
1311
"get_subarray_index",
1412
"weighted_mean_std_ufunc",
@@ -261,7 +259,7 @@ def calc_combs_min_distances_event(
261259
distances = np.hypot(lon_diffs, lat_diffs)
262260

263261
# Weighted mean for minimum distances
264-
argmin_distance = argmin(distances)
262+
argmin_distance = np.argmin(distances)
265263
lon_vals = [
266264
fov_lon_values[tel_1, sign_combs[argmin_distance, 0]],
267265
fov_lon_values[tel_2, sign_combs[argmin_distance, 1]],

0 commit comments

Comments
 (0)