seti4.py

fig, ax = plt.subplots(4, 4, figsize=(12, 10))
id1 = 0
id2 = 0
for signal_type in df['signal_classification'].unique().tolist():
    data_ids = df[df['signal_classification']==signal_type].sample(n=2)['uuid'].tolist()
    data_files = [f for f in raw_signal_files if [x for x in data_ids if x in f]]

    for idx, data_file in enumerate(data_files):
        data = ibmseti.compamp.SimCompamp(open(data_file,'rb').read())
        spectrogram = data.get_spectrogram()
        ax[id1, id2].set_title(data.header().get("signal_classification"))
        ax[id1, id2].imshow(spectrogram, aspect='auto')
        id2 += 1
        id2 %= 4
        if id2 == 0:
            id1 += 1
ax[3,2].set_axis_off()
ax[3,3].set_axis_off()
fig.tight_layout()