[~] Refactor

2021-07-29 22:26:20 +03:00 · 2021-07-29 22:26:20 +03:00 · a1dc176c32
commit a1dc176c32
parent d89b3c1c31
1 changed files with 5 additions and 1 deletions
--- a/python/tasks/mlb_player.py
+++ b/python/tasks/mlb_player.py
@ -1833,11 +1833,15 @@ def kernel_30(o_29, ids=None, delay=None,):
            ],
            axis=0
        ).prod(axis=0)
-        pprint.pprint([t8.sum(), t8.mean()])
        ids = numpy.random.choice(
            numpy.where(t8)[0],
            min(70, t8.sum()),
        )
+        pprint.pprint([
+            t8.sum(),
+            t8.mean(),
+            o_29['t5'].sel(index=o_29['t5'].index.data[ids]),
+        ])
    else:
        assert isinstance(ids, numpy.ndarray)