Commit 757748a3 authored by Marius Kriegerowski's avatar Marius Kriegerowski
Browse files

use pyrockos' argmax

parent 910837ca
......@@ -9,7 +9,7 @@ from collections import defaultdict
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 pyrocko.guts import Object, Timestamp, String, Float
from lassie import common, plot, grid as gridmod, geo
......@@ -418,9 +418,7 @@ def scan(
tr_stackmax_indx = tr_stackmax.copy(data=False)
imaxs = num.zeros(len(frames))
for iframe, frame in enumerate(frames.T):
imaxs[iframe] = num.argmax(frame)
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