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

adding plot with chirps in chasing

Browse Source
This commit is contained in:
wendtalexander 2023-01-21 10:03:57 +01:00
parent 7a86d53574
commit d425b8dafa
1 changed files with 33 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
code/behavior.py
View File

@ -157,6 +157,7 @@ def main(datapath: str):
ax1.scatter(physical_contact, np.ones_like(physical_contact)*3, marker='x', color='black', label='Physical contact') ax1.scatter(physical_contact, np.ones_like(physical_contact)*3, marker='x', color='black', label='Physical contact')
plt.legend() plt.legend()
# plt.show() # plt.show()
plt.close()
# Get fish ids # Get fish ids
all_fish_ids = np.unique(chirps_fish_ids) all_fish_ids = np.unique(chirps_fish_ids)
@ -173,14 +174,40 @@ def main(datapath: str):
ax2.bar(x, fish, width=width) ax2.bar(x, fish, width=width)
ax2.set_ylabel('Chirp count') ax2.set_ylabel('Chirp count')
# plt.show() # plt.show()
embed() plt.close()
##### Count chirps emitted during chasing events and chirps emitted out of chasing events ##### ##### Count chirps emitted during chasing events and chirps emitted out of chasing events #####
chirps_in_chasing = [] chirps_in_chasings = []
for onset, offset in zip(chasing_onset, chasing_offset): for onset, offset in zip(chasing_onset, chasing_offset):
if chirps.any((chirps > onset) & (chirps < offset)): chirps_in_chasing = [c for c in chirps if (c > onset) & (c < offset)]
chirps_in_chasing.append(chirps) chirps_in_chasings.append(chirps_in_chasing)
print(chirps_in_chasing)
embed() # chirps out of chasing events
counts_chirps_chasings = 0
chasings_without_chirps = 0
for i in chirps_in_chasings:
if i:
chasings_without_chirps += 1
else:
counts_chirps_chasings += 1
# chirps in chasing events
fig3 , ax3 = plt.subplots()
ax3.bar(['Chirps in chasing events', 'Chasing events without Chirps'], [counts_chirps_chasings, chasings_without_chirps], width=width)
plt.ylabel('Count')
plt.show()
plt.close()
# comparison between chasing events with and without chirps
embed()