import pyfar as pf
import numpy as np
signal_1 = pf.signals.impulse(5, 0)
delays = np.array([[0, 1], [2, 3]], dtype=int)
signal_2 = pf.signals.impulse(5, delays)
cor = pf.dsp.correlate(signal_1, signal_2, 'full')
argmax = cor.times[np.argmax(cor.time, axis=-1)]
ax = pf.plot.time(cor, unit='ms')
ax.set_title('Correlation and position of maxima (dots)')
ax.set_xlabel('Time lag in ms')
ax.set_ylabel('Auto correlation')
for amax, color in zip(argmax.flatten(), 'bryp'):
    ax.axvline(amax, color=pf.plot.color(color), linestyle=':')