From 988d80f7bd683c082d4a479394603e3fedeec3a9 Mon Sep 17 00:00:00 2001 From: Till Raab Date: Fri, 16 Jun 2023 15:37:54 +0200 Subject: [PATCH] work in progress --- ethogram.py | 7 ++++--- event_time_correlations.py | 9 +++++++++ 2 files changed, 13 insertions(+), 3 deletions(-) diff --git a/ethogram.py b/ethogram.py index 2479739..2b4d544 100644 --- a/ethogram.py +++ b/ethogram.py @@ -9,7 +9,7 @@ import networkx as nx from IPython import embed from event_time_correlations import load_and_converete_boris_events -glob_colors = ['#BA2D22', '#53379B', '#F47F17', '#3673A4', '#AAB71B', '#DC143C', '#1E90FF'] +glob_colors = ['#BA2D22', '#53379B', '#F47F17', '#3673A4', '#AAB71B', '#DC143C', '#1E90FF', 'k'] def plot_transition_matrix(matrix, labels): @@ -139,9 +139,10 @@ def main(base_path): for enu_tar, label_tar in enumerate(loop_labels): n = len(event_times[:-1][(event_labels[1:] == label_tar) & (np.diff(event_times) > 5)]) marcov_matrix[-1, enu_tar] = n + marcov_matrix[-1, 5] = 0 + loop_labels.append('void') + event_counts = np.append(event_counts, marcov_matrix[-1].sum()) - embed() - quit() ### get those cases where ag_on does not point to event and no event points to corresponding ag_off ... add thise cases in marcov matrix chase_on_idx = np.where(event_labels == loop_labels[4])[0] chase_off_idx = np.where(event_labels == loop_labels[5])[0] diff --git a/event_time_correlations.py b/event_time_correlations.py index b80017c..80ab703 100644 --- a/event_time_correlations.py +++ b/event_time_correlations.py @@ -255,12 +255,14 @@ def main(base_path): lose_chrips_centered_on_ag_on_t = [] lose_chrips_centered_on_contact_t = [] lose_chrips_centered_on_win_rises = [] + lose_chrips_centered_on_win_chirp = [] lose_chirp_count = [] win_chrips_centered_on_ag_off_t = [] win_chrips_centered_on_ag_on_t = [] win_chrips_centered_on_contact_t = [] win_chrips_centered_on_lose_rises = [] + win_chrips_centered_on_lose_chirp = [] win_chirp_count = [] lose_rises_centered_on_ag_off_t = [] @@ -317,24 +319,29 @@ def main(base_path): lose_chrips_centered_on_ag_on_t.append(event_centered_times(ag_on_off_t_GRID[:, 0], chirp_times[1])) lose_chrips_centered_on_contact_t.append(event_centered_times(contact_t_GRID, chirp_times[1])) lose_chrips_centered_on_win_rises.append(event_centered_times(rise_times[0], chirp_times[1])) + lose_chrips_centered_on_win_chirp.append(event_centered_times(chirp_times[0], chirp_times[1])) lose_chirp_count.append(len(chirp_times[1])) win_chrips_centered_on_ag_off_t.append(event_centered_times(ag_on_off_t_GRID[:, 1], chirp_times[0])) win_chrips_centered_on_ag_on_t.append(event_centered_times(ag_on_off_t_GRID[:, 0], chirp_times[0])) win_chrips_centered_on_contact_t.append(event_centered_times(contact_t_GRID, chirp_times[0])) win_chrips_centered_on_lose_rises.append(event_centered_times(rise_times[1], chirp_times[0])) + win_chrips_centered_on_lose_chirp.append(event_centered_times(chirp_times[1], chirp_times[0])) + win_chirp_count.append(len(chirp_times[0])) else: lose_chrips_centered_on_ag_off_t.append(np.array([])) lose_chrips_centered_on_ag_on_t.append(np.array([])) lose_chrips_centered_on_contact_t.append(np.array([])) lose_chrips_centered_on_win_rises.append(np.array([])) + lose_chrips_centered_on_win_chirp.append(np.array([])) lose_chirp_count.append(np.nan) win_chrips_centered_on_ag_off_t.append(np.array([])) win_chrips_centered_on_ag_on_t.append(np.array([])) win_chrips_centered_on_contact_t.append(np.array([])) win_chrips_centered_on_lose_rises.append(np.array([])) + win_chrips_centered_on_lose_chirp.append(np.array([])) win_chirp_count.append(np.nan) # rises @@ -369,11 +376,13 @@ def main(base_path): [lose_chrips_centered_on_ag_on_t, lose_chirp_count, r'chirp$_{lose}$ on chase$_{on}$'], [lose_chrips_centered_on_contact_t, lose_chirp_count, r'chirp$_{lose}$ on contact'], [lose_chrips_centered_on_win_rises, lose_chirp_count, r'chirp$_{lose}$ on rise$_{win}$'], + [lose_chrips_centered_on_win_chirp, lose_chirp_count, r'chirp$_{lose}$ on chirp$_{win}$'], [win_chrips_centered_on_ag_off_t, win_chirp_count, r'chirp$_{win}$ on chase$_{off}$'], [win_chrips_centered_on_ag_on_t, win_chirp_count, r'chirp$_{win}$ on chase$_{on}$'], [win_chrips_centered_on_contact_t, win_chirp_count, r'chirp$_{win}$ on contact'], [win_chrips_centered_on_lose_rises, win_chirp_count, r'chirp$_{win}$ on rise$_{lose}$'], + [win_chrips_centered_on_lose_chirp, win_chirp_count, r'chirp$_{win}$ on chirp$_{lose}$'], [lose_rises_centered_on_ag_off_t, lose_rises_count, r'rise$_{lose}$ on chase$_{off}$'], [lose_rises_centered_on_ag_on_t, lose_rises_count, r'rise$_{lose}$ on chase$_{on}$'],