raab/GP2023_chirp_detection
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

finish function

Browse Source
This commit is contained in:
wendtalexander 2023-01-25 13:59:13 +01:00
parent a5ae0a3032
commit cdcf9564df
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
code/plot_chirp_size.py
View File

@ -65,24 +65,27 @@ def get_chirp_size(folder_name, Behavior, order_meta_df, id_meta_df):
mean_size_loser = np.nanmean(size_losers) mean_size_loser = np.nanmean(size_losers)
if mean_size_winner > mean_size_loser: if mean_size_winner > mean_size_loser:
size_diff = mean_size_winner - mean_size_loser size_diff_bigger = mean_size_winner - mean_size_loser
size_diff_smaller = mean_size_loser - mean_size_winner
winner_fish_id = folder_row['rec_id1'].values[0] winner_fish_id = folder_row['rec_id1'].values[0]
loser_fish_id = folder_row['rec_id2'].values[0] loser_fish_id = folder_row['rec_id2'].values[0]
elif mean_size_winner < mean_size_loser: elif mean_size_winner < mean_size_loser:
size_diff = mean_size_loser - mean_size_winner size_diff_bigger = mean_size_loser - mean_size_winner
size_diff_smaller = mean_size_winner - mean_size_loser
winner_fish_id = folder_row['rec_id2'].values[0] winner_fish_id = folder_row['rec_id2'].values[0]
loser_fish_id = folder_row['rec_id1'].values[0] loser_fish_id = folder_row['rec_id1'].values[0]
else: else:
size_diff = np.nan size_diff = np.nan
winner_fish_id = np.nan winner_fish_id = np.nan
loser_fish_id = np.nan loser_fish_id = np.nan
chirp_diff = len(Behavior.chirps[Behavior.chirps_ids == winner_fish_id]) - len( chirp_winner = len(
Behavior.chirps[Behavior.chirps_ids == winner_fish_id])
chirp_loser = len(
Behavior.chirps[Behavior.chirps_ids == loser_fish_id]) Behavior.chirps[Behavior.chirps_ids == loser_fish_id])
return size_diff, chirp_diff return size_diff_bigger, chirp_winner, size_diff_smaller, chirp_loser
def get_chirp_freq(folder_name, Behavior, order_meta_df): def get_chirp_freq(folder_name, Behavior, order_meta_df):
@ -190,7 +193,7 @@ def main(datapath: str):
ax2.set_xlabel('Size difference [mm]') ax2.set_xlabel('Size difference [mm]')
ax2.set_ylabel('Chirps difference [n]') ax2.set_ylabel('Chirps difference [n]')
ax3.scatter(freq_diffs, size_chirps_diffs, color='r') #ax3.scatter(freq_diffs, size_chirps_diffs, color='r')
# ax3.scatter(freq_diffs, freq_chirps_diffs, color='r') # ax3.scatter(freq_diffs, freq_chirps_diffs, color='r')
ax3.set_xlabel('Frequency difference [Hz]') ax3.set_xlabel('Frequency difference [Hz]')
ax3.set_yticklabels([]) ax3.set_yticklabels([])