Commit aa567793 authored by Marius Kriegerowski's avatar Marius Kriegerowski
Browse files

use pyrockos argmax

parent 8599354b
......@@ -7,7 +7,7 @@ import shutil
import numpy as num
from pyrocko import pile, trace, util, io
from pyrocko.parstack import parstack
from pyrocko.parstack import parstack, argmax as pargmax
from lassie import common, plot, grid as gridmod
......@@ -232,7 +232,7 @@ def scan(
config.detector_threshold, shift_max + 1.0/fsmooth_min)
tr_stackmax_indx = tr_stackmax.copy(data=False)
imaxs = num.argmax(frames, axis=0)
imaxs = pargmax(frames)
tr_stackmax_indx.set_ydata(imaxs.astype(num.int32))
tr_stackmax_indx.set_location('i')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment