Skip to content

Commit

Permalink
getting a column as dict by index is now removed from 'sparse numeric…
Browse files Browse the repository at this point in the history
… table'.
  • Loading branch information
relleums committed Aug 22, 2024
1 parent 910f1a2 commit 2d6d9cc
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 2 deletions.
11 changes: 10 additions & 1 deletion plenoirf/production/estimate_primary_trajectory.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,6 +53,14 @@ def run_block(env, blk, block_id, logger):
logger.info(__name__ + ": ... done.")


def get_column_as_dict_by_index(table, level_key, column_key, index_key):
level = table[level_key]
out = {}
for ii in range(level.shape[0]):
out[level[index_key][ii]] = level[column_key][ii]
return out


def estimate_primary_trajectory(
evttab,
fuzzy_config,
Expand All @@ -61,10 +69,11 @@ def estimate_primary_trajectory(
light_field_geometry,
logger,
):
shower_maximum_object_distance = snt.get_column_as_dict_by_index(
shower_maximum_object_distance = get_column_as_dict_by_index(
table=evttab,
level_key="features",
column_key="image_smallest_ellipse_object_distance",
index_key="idx",
)

run = plenopy.photon_stream.loph.LopfTarReader(
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -137,7 +137,7 @@ def read_shower_maximum_object_distance(
structure=irf.table.STRUCTURE,
)

return snt.get_column_as_dict_by_index(
return irf.production.estimate_primary_trajectory.get_column_as_dict_by_index(
table=event_table, level_key="features", column_key=key
)

Expand Down

0 comments on commit 2d6d9cc

Please sign in to comment.