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

adding variable for scatter plot

Browse Source
This commit is contained in:
wendtalexander 2023-01-24 14:07:38 +01:00
parent 726301c593
commit a371755b1c
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
code/plot_chirp_bodylegth.py
View File

@ -69,16 +69,18 @@ def main(datapath: str):
print(all_fish_ids) print(all_fish_ids)
fig, ax = plt.subplots() fig, ax = plt.subplots()
scatterwinner = 1.15
scatterloser = 1.85
bplot1 = ax.boxplot(chirps_winner, positions=[ bplot1 = ax.boxplot(chirps_winner, positions=[
1], showfliers=False, patch_artist=True) 1], showfliers=False, patch_artist=True)
bplot2 = ax.boxplot(chirps_loser, positions=[ bplot2 = ax.boxplot(chirps_loser, positions=[
2], showfliers=False, patch_artist=True) 2], showfliers=False, patch_artist=True)
ax.scatter(np.ones(len(chirps_winner))*1.15, chirps_winner, color='r') ax.scatter(np.ones(len(chirps_winner))*scatterwinner, chirps_winner, color='r')
ax.scatter(np.ones(len(chirps_loser))*1.85, chirps_loser, color='r') ax.scatter(np.ones(len(chirps_loser))*scatterloser, chirps_loser, color='r')
ax.set_xticklabels(['winner', 'loser']) ax.set_xticklabels(['winner', 'loser'])
for w, l in zip(chirps_winner, chirps_loser): for w, l in zip(chirps_winner, chirps_loser):
ax.plot([1.15, 1.85], [w, l], color='r', alpha=0.5, linewidth=0.5) ax.plot([scatterwinner, scatterloser], [w, l], color='r', alpha=0.5, linewidth=0.5)
colors1 = ps.red colors1 = ps.red
ps.set_boxplot_color(bplot1, colors1) ps.set_boxplot_color(bplot1, colors1)
@ -87,6 +89,7 @@ def main(datapath: str):
ax.set_ylabel('Chirpscounts [n]') ax.set_ylabel('Chirpscounts [n]')
plt.savefig('../poster/figs/chirps_winner_loser.pdf') plt.savefig('../poster/figs/chirps_winner_loser.pdf')
plt.show()
if __name__ == '__main__': if __name__ == '__main__':