From 8c1f85f89e5f3b4cec689df2117ed59a9f70d8eb Mon Sep 17 00:00:00 2001 From: Till Raab Date: Tue, 23 May 2023 11:01:59 +0200 Subject: [PATCH] cleaned code --- complete_analysis.py | 7 +- trial_summary.csv | 10 +- trial_summary_eval.py | 392 ++++++++++++++++++++++++++++-------------- 3 files changed, 268 insertions(+), 141 deletions(-) diff --git a/complete_analysis.py b/complete_analysis.py index 5640373..9c2c25b 100644 --- a/complete_analysis.py +++ b/complete_analysis.py @@ -181,7 +181,7 @@ def main(data_folder=None): trial_summary = pd.DataFrame(columns=['recording', 'group', 'win_fish', 'lose_fish', 'sex_win', 'sex_lose', 'size_win', 'size_lose', 'EODf_win', 'EODf_lose', 'exp_win', 'exp_lose', - 'chirps_win', 'chirps_lose', 'rises_win', 'rise_lose', 'draw']) + 'chirps_win', 'chirps_lose', 'rises_win', 'rises_lose', 'draw']) trial_summary_row = {f'{s}':None for s in trial_summary.keys()} for trial_idx in tqdm(np.arange(len(trials_meta)), desc='Trials'): @@ -299,7 +299,7 @@ def main(data_folder=None): 'chirps_win': len(chirp_times[0]), 'chirps_lose': len(chirp_times[1]), 'rises_win': len(rise_idx_int[0]), - 'rise_lose': len(rise_idx_int[1]), + 'rises_lose': len(rise_idx_int[1]), 'draw': 1 if trials_meta['winner'][trial_idx] == -1 else 0 } # embed() @@ -393,8 +393,7 @@ def main(data_folder=None): sex = 'm' trial_summary['sex_win'][(trial_summary['group'] == g) & (trial_summary['win_fish'] == f)] = sex trial_summary['sex_lose'][(trial_summary['group'] == g) & (trial_summary['lose_fish'] == f)] = sex - embed() - quit() + trial_summary.to_csv('trial_summary.csv') pass if __name__ == '__main__': diff --git a/trial_summary.csv b/trial_summary.csv index ea67eb0..fadad90 100644 --- a/trial_summary.csv +++ b/trial_summary.csv @@ -1,4 +1,4 @@ -,recording,group,win_fish,lose_fish,sex_win,sex_lose,size_win,size_lose,EODf_win,EODf_lose,exp_win,exp_lose,chirps_win,chirps_lose,rises_win,rise_lose,draw +,recording,group,win_fish,lose_fish,sex_win,sex_lose,size_win,size_lose,EODf_win,EODf_lose,exp_win,exp_lose,chirps_win,chirps_lose,rises_win,rises_lose,draw 0,2019-11-25-09_59,3,1,2,f,f,13.2,12.0,713.0544113886845,762.0273047058653,1,1,36,2657,22,165,0 1,2019-11-26-10_00,3,4,3,m,m,15.5,17.5,883.141322780704,918.0584506431281,1,1,472,1322,17,481,0 2,2019-11-27-10_00,3,5,6,f,f,14.4,12.65,728.1663791991439,650.6079943890219,1,1,16,2041,14,311,0 @@ -13,14 +13,14 @@ 11,2019-12-11-10_00,3,4,1,m,f,15.5,13.2,909.5564241038855,704.759181051688,4,5,44,260,48,165,0 12,2019-12-12-10_00,3,2,6,f,f,12.0,12.65,708.2029632781753,649.3215729301896,5,5,55,1489,26,152,0 13,2019-12-16-10_00,3,4,5,m,f,15.5,14.4,911.4475182245616,734.3463774893517,5,5,52,963,39,123,0 -14,2020-03-13-10_00,4,5,4,m,f,12.5,12.266666666666666,726.3470010966499,705.1654694195288,2,2,54,941,70,177,0 +14,2020-03-13-10_00,4,5,4,f,f,12.5,12.266666666666666,726.3470010966499,705.1654694195288,2,2,54,941,70,177,0 15,2020-03-16-10_00,4,3,1,m,f,11.933333333333332,11.299999999999999,852.2318545355058,642.0347177867645,3,3,1304,724,57,154,0 -16,2020-03-18-10_34,4,5,3,m,m,12.5,11.933333333333332,725.8257351336636,863.6524533012707,3,4,16557,2089,339,43,1 +16,2020-03-18-10_34,4,5,3,f,m,12.5,11.933333333333332,725.8257351336636,863.6524533012707,3,4,16557,2089,339,43,1 17,2020-03-19-10_00,4,1,4,f,f,11.299999999999999,12.266666666666666,659.5490944255365,697.5034008357667,4,4,52,1583,36,197,0 -18,2020-03-20-10_00,4,5,2,m,f,12.5,12.266666666666666,,,4,4,45,665,75,76,0 +18,2020-03-20-10_00,4,5,2,f,f,12.5,12.266666666666666,,,4,4,45,665,75,76,0 19,2020-03-23-09_58,4,2,1,f,f,12.266666666666666,11.299999999999999,699.4914052830558,654.7533296886725,5,5,84,1158,17,67,1 20,2020-03-24-10_00,4,4,3,f,m,12.266666666666666,11.933333333333332,684.578069899078,854.0458114588357,5,5,883,2,184,86,1 -21,2020-03-25-10_00,4,5,1,m,f,12.5,11.299999999999999,733.5001619575638,647.9874053272127,5,6,819,1831,48,70,1 +21,2020-03-25-10_00,4,5,1,f,f,12.5,11.299999999999999,733.5001619575638,647.9874053272127,5,6,819,1831,48,70,1 22,2020-03-31-09_59,4,3,2,m,f,11.933333333333332,12.266666666666666,860.5459022492297,692.2978867242133,6,6,10,225,26,50,1 23,2020-05-11-10_00,5,1,2,m,f,12.333333333333334,13.166666666666666,834.369973908149,667.9762847453638,1,1,4,631,25,230,0 24,2020-05-12-10_00,5,5,3,f,m,19.0,10.666666666666666,697.6088902440882,818.2108387976053,1,1,1,117,8,429,0 diff --git a/trial_summary_eval.py b/trial_summary_eval.py index 95654ab..8943201 100644 --- a/trial_summary_eval.py +++ b/trial_summary_eval.py @@ -16,7 +16,7 @@ def plot_rise_vs_chirp_count(trial_summary): ax = fig.add_subplot(gs[1, 0]) ax.plot(trial_summary['rises_win'], trial_summary['chirps_win'], 'o', color=Wc, label='winner') - ax.plot(trial_summary['rise_lose'], trial_summary['chirps_lose'], 'o', color=Lc, label='loster') + ax.plot(trial_summary['rises_lose'], trial_summary['chirps_lose'], 'o', color=Lc, label='loster') ax.set_xlabel('rises [n]', fontsize=12) ax.set_ylabel('chirps [n]', fontsize=12) ax.tick_params(labelsize=10) @@ -28,130 +28,134 @@ def plot_rise_vs_chirp_count(trial_summary): plt.setp(ax_chirps.get_yticklabels(), visible=False) ax_rises = fig.add_subplot(gs[0, 0], sharex=ax) - ax_rises.boxplot([trial_summary['rises_win'], trial_summary['rise_lose']], widths = .5, positions = [1, 2], vert=False) + ax_rises.boxplot([trial_summary['rises_win'], trial_summary['rises_lose']], widths = .5, positions = [1, 2], vert=False) ax_rises.set_yticks([1, 2]) ax_rises.set_yticklabels(['Win', 'Lose']) plt.setp(ax_rises.get_xticklabels(), visible=False) -def plot_chirp_rise_count_per_pairing(trial_summary): - win_chirps = [] - lose_chirps = [] +def plot_chirp_rise_count_per_pairing(trial_summary, + beh_key_win=None, beh_key_lose=None, + ylabel='y'): - win_rises = [] - lose_rises = [] + mek = ['k', 'None', 'None', 'k'] + markersize = 12 + win_colors = [male_color, male_color, female_color, female_color] + lose_colors = [male_color, female_color, male_color, female_color] + + win_count = [] + lose_count = [] for win_sex, lose_sex in itertools.product(['m', 'f'], repeat=2): - win_chirps.append(trial_summary['chirps_win'][(trial_summary["sex_win"] == win_sex) & + win_count.append(trial_summary[beh_key_win][(trial_summary["sex_win"] == win_sex) & (trial_summary["sex_lose"] == lose_sex) & (trial_summary["draw"] == 0)].to_numpy()) - lose_chirps.append(trial_summary['chirps_lose'][(trial_summary["sex_win"] == win_sex) & + lose_count.append(trial_summary[beh_key_lose][(trial_summary["sex_win"] == win_sex) & (trial_summary["sex_lose"] == lose_sex) & (trial_summary["draw"] == 0)].to_numpy()) - win_rises.append(trial_summary['rises_win'][(trial_summary["sex_win"] == win_sex) & - (trial_summary["sex_lose"] == lose_sex) & - (trial_summary["draw"] == 0)].to_numpy()) - lose_rises.append(trial_summary['rise_lose'][(trial_summary["sex_win"] == win_sex) & - (trial_summary["sex_lose"] == lose_sex) & - (trial_summary["draw"] == 0)].to_numpy()) - - - fig = plt.figure(figsize=(20/2.54, 12/2.54)) - gs = gridspec.GridSpec(1, 1, left=0.1, bottom=0.1, right=0.95, top=0.95) - ax = fig.add_subplot(gs[0, 0]) - ax.boxplot(win_chirps, positions=np.arange(len(win_chirps))-0.15, widths= .2, sym='') - ax.boxplot(lose_chirps, positions=np.arange(len(lose_chirps))+0.15, widths= .2, sym='') - - ax.set_xticks(np.arange(len(win_chirps))) - ax.set_xticklabels([u'\u2642\u2642', u'\u2642\u2640', u'\u2640\u2642', u'\u2640\u2640']) - # ax.set_xticklabels(['mm', 'mf', 'fm', 'ff']) - y0, y1 = ax.get_ylim() - for i in range(len(win_chirps)): - ax.text(i, y1, f'n={len(win_chirps[i]):.0f}', fontsize=10, ha='center', va='bottom') - ax.set_ylim(top = y1*1.1) - ax.set_ylabel('chirps [n]', fontsize=12) - plt.tick_params(labelsize=10) fig = plt.figure(figsize=(20/2.54, 12/2.54)) gs = gridspec.GridSpec(1, 1, left=0.1, bottom=0.1, right=0.95, top=0.95) ax = fig.add_subplot(gs[0, 0]) - ax.boxplot(win_rises, positions=np.arange(len(win_rises))-0.15, widths= .2, sym='') - ax.boxplot(lose_rises, positions=np.arange(len(lose_rises))+0.15, widths= .2, sym='') + ax.boxplot(win_count, positions=np.arange(len(win_count))-0.15, widths= .2, sym='') + ax.boxplot(lose_count, positions=np.arange(len(lose_count))+0.15, widths= .2, sym='') - ax.set_xticks(np.arange(len(win_rises))) + ax.set_xticks(np.arange(len(win_count))) ax.set_xticklabels([u'\u2642\u2642', u'\u2642\u2640', u'\u2640\u2642', u'\u2640\u2640']) # ax.set_xticklabels(['mm', 'mf', 'fm', 'ff']) y0, y1 = ax.get_ylim() - for i in range(len(win_rises)): - ax.text(i, y1, f'n={len(win_rises[i]):.0f}', fontsize=10, ha='center', va='bottom') + for i in range(len(win_count)): + ax.text(i, y1, f'n={len(win_count[i]):.0f}', fontsize=10, ha='center', va='bottom') ax.set_ylim(top = y1*1.1) - ax.set_ylabel('rises [n]', fontsize=12) + ax.set_ylabel(ylabel, fontsize=12) plt.tick_params(labelsize=10) - # plt.show() + # win_chirps = [] + # lose_chirps = [] + # + # win_rises = [] + # lose_rises = [] + # + # for win_sex, lose_sex in itertools.product(['m', 'f'], repeat=2): + # win_chirps.append(trial_summary['chirps_win'][(trial_summary["sex_win"] == win_sex) & + # (trial_summary["sex_lose"] == lose_sex) & + # (trial_summary["draw"] == 0)].to_numpy()) + # lose_chirps.append(trial_summary['chirps_lose'][(trial_summary["sex_win"] == win_sex) & + # (trial_summary["sex_lose"] == lose_sex) & + # (trial_summary["draw"] == 0)].to_numpy()) + # win_rises.append(trial_summary['rises_win'][(trial_summary["sex_win"] == win_sex) & + # (trial_summary["sex_lose"] == lose_sex) & + # (trial_summary["draw"] == 0)].to_numpy()) + # lose_rises.append(trial_summary['rises_lose'][(trial_summary["sex_win"] == win_sex) & + # (trial_summary["sex_lose"] == lose_sex) & + # (trial_summary["draw"] == 0)].to_numpy()) + # + # + # fig = plt.figure(figsize=(20/2.54, 12/2.54)) + # gs = gridspec.GridSpec(1, 1, left=0.1, bottom=0.1, right=0.95, top=0.95) + # ax = fig.add_subplot(gs[0, 0]) + # ax.boxplot(win_chirps, positions=np.arange(len(win_chirps))-0.15, widths= .2, sym='') + # ax.boxplot(lose_chirps, positions=np.arange(len(lose_chirps))+0.15, widths= .2, sym='') + # + # ax.set_xticks(np.arange(len(win_chirps))) + # ax.set_xticklabels([u'\u2642\u2642', u'\u2642\u2640', u'\u2640\u2642', u'\u2640\u2640']) + # # ax.set_xticklabels(['mm', 'mf', 'fm', 'ff']) + # y0, y1 = ax.get_ylim() + # for i in range(len(win_chirps)): + # ax.text(i, y1, f'n={len(win_chirps[i]):.0f}', fontsize=10, ha='center', va='bottom') + # ax.set_ylim(top = y1*1.1) + # ax.set_ylabel('chirps [n]', fontsize=12) + # plt.tick_params(labelsize=10) + # + # fig = plt.figure(figsize=(20/2.54, 12/2.54)) + # gs = gridspec.GridSpec(1, 1, left=0.1, bottom=0.1, right=0.95, top=0.95) + # ax = fig.add_subplot(gs[0, 0]) + # ax.boxplot(win_rises, positions=np.arange(len(win_rises))-0.15, widths= .2, sym='') + # ax.boxplot(lose_rises, positions=np.arange(len(lose_rises))+0.15, widths= .2, sym='') + # + # ax.set_xticks(np.arange(len(win_rises))) + # ax.set_xticklabels([u'\u2642\u2642', u'\u2642\u2640', u'\u2640\u2642', u'\u2640\u2640']) + # # ax.set_xticklabels(['mm', 'mf', 'fm', 'ff']) + # y0, y1 = ax.get_ylim() + # for i in range(len(win_rises)): + # ax.text(i, y1, f'n={len(win_rises[i]):.0f}', fontsize=10, ha='center', va='bottom') + # ax.set_ylim(top = y1*1.1) + # ax.set_ylabel('rises [n]', fontsize=12) + # plt.tick_params(labelsize=10) + # # plt.show() + + +def plot_beh_count_vs_meta(trial_summary, + beh_key_win=None, beh_key_lose=None, + meta_key_win=None, meta_key_lose=None, + xlabel='x'): + mek = ['k', 'None', 'None', 'k'] + markersize = 12 + win_colors = [male_color, male_color, female_color, female_color] + lose_colors = [male_color, female_color, male_color, female_color] -def plot_chirp_rise_count_vs_size_diff(trial_summary): - win_chirps = [] - lose_chirps = [] - win_rises = [] - lose_rises = [] + win_count = [] + lose_count = [] - d_size = [] - d_EODf = [] + win_meta = [] + lose_meta = [] for win_sex, lose_sex in itertools.product(['m', 'f'], repeat=2): - win_chirps.append(trial_summary['chirps_win'][(trial_summary["sex_win"] == win_sex) & + win_count.append(trial_summary[beh_key_win][(trial_summary["sex_win"] == win_sex) & (trial_summary["sex_lose"] == lose_sex) & (trial_summary["draw"] == 0)].to_numpy()) - lose_chirps.append(trial_summary['chirps_lose'][(trial_summary["sex_win"] == win_sex) & + lose_count.append(trial_summary[beh_key_lose][(trial_summary["sex_win"] == win_sex) & (trial_summary["sex_lose"] == lose_sex) & (trial_summary["draw"] == 0)].to_numpy()) - win_rises.append(trial_summary['rises_win'][(trial_summary["sex_win"] == win_sex) & - (trial_summary["sex_lose"] == lose_sex) & - (trial_summary["draw"] == 0)].to_numpy()) - lose_rises.append(trial_summary['rise_lose'][(trial_summary["sex_win"] == win_sex) & - (trial_summary["sex_lose"] == lose_sex) & - (trial_summary["draw"] == 0)].to_numpy()) - - w_size = trial_summary['size_win'][(trial_summary["sex_win"] == win_sex) & - (trial_summary["sex_lose"] == lose_sex) & - (trial_summary["draw"] == 0)].to_numpy() - l_size = trial_summary['size_lose'][(trial_summary["sex_win"] == win_sex) & - (trial_summary["sex_lose"] == lose_sex) & - (trial_summary["draw"] == 0)].to_numpy() - w_EODf = trial_summary['EODf_win'][(trial_summary["sex_win"] == win_sex) & + win_meta.append(trial_summary[meta_key_win][(trial_summary["sex_win"] == win_sex) & (trial_summary["sex_lose"] == lose_sex) & - (trial_summary["draw"] == 0)].to_numpy() - l_EODf = trial_summary['EODf_lose'][(trial_summary["sex_win"] == win_sex) & + (trial_summary["draw"] == 0)].to_numpy()) + lose_meta.append(trial_summary[meta_key_lose][(trial_summary["sex_win"] == win_sex) & (trial_summary["sex_lose"] == lose_sex) & - (trial_summary["draw"] == 0)].to_numpy() - - d_size.append(w_size-l_size) - d_EODf.append(w_EODf-l_EODf) - - - mek = ['k', 'None', 'None', 'k'] - markersize = 12 - win_colors = [male_color, male_color, female_color, female_color] - lose_colors = [male_color, female_color, male_color, female_color] - - ### SIZE - ################################################################################################################## - fig = plt.figure(figsize=(20/2.54, 12/2.54)) - gs = gridspec.GridSpec(1, 1, left=0.1, bottom=0.15, right=0.95, top=0.95) - ax = fig.add_subplot(gs[0, 0]) - - for i in range(len(lose_rises)): - ax.plot(d_size[i], lose_rises[i], 'p', color=win_colors[i], markeredgecolor=mek[i], markersize=markersize) - # ax.plot(d_size[i]*-1, lose_rises[i], 'o', color=lose_colors[i], markeredgecolor=mek[i], markersize=markersize) - - ax.set_ylabel('lose rises [n]', fontsize=12) - ax.set_xlabel(u'$\Delta$size [cm]', fontsize=12) - plt.tick_params(labelsize=10) + (trial_summary["draw"] == 0)].to_numpy()) - ################################################################################################################## fig = plt.figure(figsize=(20/2.54, 20/2.54)) gs = gridspec.GridSpec(2, 2, left=0.1, bottom=0.1, right=0.95, top=0.95, hspace=0.1, wspace=0.1) ax = [] @@ -160,17 +164,17 @@ def plot_chirp_rise_count_vs_size_diff(trial_summary): ax.append(fig.add_subplot(gs[0, 1], sharey=ax[0])) ax.append(fig.add_subplot(gs[1, 1], sharex=ax[2], sharey=ax[1])) - for i in range(len(lose_chirps)): - ax[0].plot(d_size[i], win_chirps[i], 'p', color=win_colors[i], markeredgecolor=mek[i], markersize=markersize, markeredgewidth=2) - ax[1].plot(d_size[i], lose_chirps[i], 'p', color=win_colors[i], markeredgecolor=mek[i], markersize=markersize, markeredgewidth=2) + for i in range(len(win_count)): + ax[0].plot(win_meta[i]-lose_meta[i], win_count[i], 'p', color=win_colors[i], markeredgecolor=mek[i], markersize=markersize, markeredgewidth=2) + ax[1].plot(win_meta[i]-lose_meta[i], lose_count[i], 'p', color=win_colors[i], markeredgecolor=mek[i], markersize=markersize, markeredgewidth=2) - ax[2].plot(d_size[i], win_chirps[i], 'o', color=lose_colors[i], markeredgecolor=mek[i], markersize=markersize, markeredgewidth=2) - ax[3].plot(d_size[i], lose_chirps[i], 'o', color=lose_colors[i], markeredgecolor=mek[i], markersize=markersize, markeredgewidth=2 ) + ax[2].plot((win_meta[i]-lose_meta[i])*-1, win_count[i], 'o', color=lose_colors[i], markeredgecolor=mek[i], markersize=markersize, markeredgewidth=2) + ax[3].plot((win_meta[i]-lose_meta[i])*-1, lose_count[i], 'o', color=lose_colors[i], markeredgecolor=mek[i], markersize=markersize, markeredgewidth=2 ) - ax[0].set_ylabel('win chirps [n]', fontsize=12) - ax[1].set_ylabel('lose chirps [n]', fontsize=12) - ax[1].set_xlabel(u'$\Delta$size [cm]', fontsize=12) - ax[3].set_xlabel(u'$\Delta$size [cm]', fontsize=12) + ax[0].set_ylabel(f'{beh_key_win} [n]', fontsize=12) + ax[1].set_ylabel(f'{beh_key_lose} [n]', fontsize=12) + ax[1].set_xlabel(f'{xlabel}', fontsize=12) + ax[3].set_xlabel(f'{xlabel}', fontsize=12) plt.setp(ax[0].get_xticklabels(), visible=False) plt.setp(ax[2].get_xticklabels(), visible=False) @@ -179,49 +183,173 @@ def plot_chirp_rise_count_vs_size_diff(trial_summary): plt.setp(ax[3].get_yticklabels(), visible=False) plt.tick_params(labelsize=10) - ### EODf - ################################################################################################################## - - fig = plt.figure(figsize=(20/2.54, 12/2.54)) - gs = gridspec.GridSpec(1, 1, left=0.1, bottom=0.15, right=0.95, top=0.95) - ax = fig.add_subplot(gs[0, 0]) - - for i in range(len(lose_rises)): - ax.plot(d_EODf[i], lose_rises[i], 'p', color=win_colors[i], markeredgecolor=mek[i], markersize=markersize) - # ax.plot(d_EODf[i]*-1, lose_rises[i], 'o', color=lose_colors[i], markeredgecolor=mek[i], markersize=markersize) - - ax.set_ylabel('lose rises [n]', fontsize=12) - ax.set_xlabel(u'$\Delta$EODf [Hz]', fontsize=12) - plt.tick_params(labelsize=10) - - ################################################################################################################## - fig = plt.figure(figsize=(20/2.54, 12/2.54)) - gs = gridspec.GridSpec(1, 1, left=0.1, bottom=0.15, right=0.95, top=0.95) - ax = fig.add_subplot(gs[0, 0]) - - for i in range(len(lose_chirps)): - ax.plot(d_EODf[i], lose_chirps[i], 'p', color=win_colors[i], markeredgecolor=mek[i], markersize=markersize) - # ax.plot(d_EODf[i]*-1, lose_chirps[i], 'o', color=lose_colors[i], markeredgecolor=mek[i], markersize=markersize) - - ax.set_ylabel('lose chirps [n]', fontsize=12) - ax.set_xlabel(u'$\Delta$EODf [Hz]', fontsize=12) - plt.tick_params(labelsize=10) + ############################################################################################################### + + # win_chirps = [] + # lose_chirps = [] + # + # win_rises = [] + # lose_rises = [] + # + # d_size = [] + # d_EODf = [] + # + # for win_sex, lose_sex in itertools.product(['m', 'f'], repeat=2): + # win_chirps.append(trial_summary['chirps_win'][(trial_summary["sex_win"] == win_sex) & + # (trial_summary["sex_lose"] == lose_sex) & + # (trial_summary["draw"] == 0)].to_numpy()) + # lose_chirps.append(trial_summary['chirps_lose'][(trial_summary["sex_win"] == win_sex) & + # (trial_summary["sex_lose"] == lose_sex) & + # (trial_summary["draw"] == 0)].to_numpy()) + # win_rises.append(trial_summary['rises_win'][(trial_summary["sex_win"] == win_sex) & + # (trial_summary["sex_lose"] == lose_sex) & + # (trial_summary["draw"] == 0)].to_numpy()) + # lose_rises.append(trial_summary['rises_lose'][(trial_summary["sex_win"] == win_sex) & + # (trial_summary["sex_lose"] == lose_sex) & + # (trial_summary["draw"] == 0)].to_numpy()) + # + # w_size = trial_summary['size_win'][(trial_summary["sex_win"] == win_sex) & + # (trial_summary["sex_lose"] == lose_sex) & + # (trial_summary["draw"] == 0)].to_numpy() + # l_size = trial_summary['size_lose'][(trial_summary["sex_win"] == win_sex) & + # (trial_summary["sex_lose"] == lose_sex) & + # (trial_summary["draw"] == 0)].to_numpy() + # + # w_EODf = trial_summary['EODf_win'][(trial_summary["sex_win"] == win_sex) & + # (trial_summary["sex_lose"] == lose_sex) & + # (trial_summary["draw"] == 0)].to_numpy() + # l_EODf = trial_summary['EODf_lose'][(trial_summary["sex_win"] == win_sex) & + # (trial_summary["sex_lose"] == lose_sex) & + # (trial_summary["draw"] == 0)].to_numpy() + # + # d_size.append(w_size-l_size) + # d_EODf.append(w_EODf-l_EODf) + # + # + # mek = ['k', 'None', 'None', 'k'] + # markersize = 12 + # win_colors = [male_color, male_color, female_color, female_color] + # lose_colors = [male_color, female_color, male_color, female_color] + # + # ### SIZE + # ################################################################################################################## + # fig = plt.figure(figsize=(20/2.54, 20/2.54)) + # gs = gridspec.GridSpec(2, 2, left=0.1, bottom=0.1, right=0.95, top=0.95, hspace=0.1, wspace=0.1) + # ax = [] + # ax.append(fig.add_subplot(gs[0, 0])) + # ax.append(fig.add_subplot(gs[1, 0], sharex=ax[0])) + # ax.append(fig.add_subplot(gs[0, 1], sharey=ax[0])) + # ax.append(fig.add_subplot(gs[1, 1], sharex=ax[2], sharey=ax[1])) + # + # for i in range(len(lose_chirps)): + # ax[0].plot(d_size[i], win_rises[i], 'p', color=win_colors[i], markeredgecolor=mek[i], markersize=markersize, markeredgewidth=2) + # ax[1].plot(d_size[i], lose_rises[i], 'p', color=win_colors[i], markeredgecolor=mek[i], markersize=markersize, markeredgewidth=2) + # + # ax[2].plot(d_size[i], win_rises[i], 'o', color=lose_colors[i], markeredgecolor=mek[i], markersize=markersize, markeredgewidth=2) + # ax[3].plot(d_size[i], lose_rises[i], 'o', color=lose_colors[i], markeredgecolor=mek[i], markersize=markersize, markeredgewidth=2 ) + # + # ax[0].set_ylabel('win rises [n]', fontsize=12) + # ax[1].set_ylabel('lose rises [n]', fontsize=12) + # ax[1].set_xlabel(u'$\Delta$size [cm]', fontsize=12) + # ax[3].set_xlabel(u'$\Delta$size [cm]', fontsize=12) + # + # plt.setp(ax[0].get_xticklabels(), visible=False) + # plt.setp(ax[2].get_xticklabels(), visible=False) + # + # plt.setp(ax[2].get_yticklabels(), visible=False) + # plt.setp(ax[3].get_yticklabels(), visible=False) + # plt.tick_params(labelsize=10) + # + # ################################################################################################################## + # fig = plt.figure(figsize=(20/2.54, 20/2.54)) + # gs = gridspec.GridSpec(2, 2, left=0.1, bottom=0.1, right=0.95, top=0.95, hspace=0.1, wspace=0.1) + # ax = [] + # ax.append(fig.add_subplot(gs[0, 0])) + # ax.append(fig.add_subplot(gs[1, 0], sharex=ax[0])) + # ax.append(fig.add_subplot(gs[0, 1], sharey=ax[0])) + # ax.append(fig.add_subplot(gs[1, 1], sharex=ax[2], sharey=ax[1])) + # + # for i in range(len(lose_chirps)): + # ax[0].plot(d_size[i], win_chirps[i], 'p', color=win_colors[i], markeredgecolor=mek[i], markersize=markersize, markeredgewidth=2) + # ax[1].plot(d_size[i], lose_chirps[i], 'p', color=win_colors[i], markeredgecolor=mek[i], markersize=markersize, markeredgewidth=2) + # + # ax[2].plot(d_size[i], win_chirps[i], 'o', color=lose_colors[i], markeredgecolor=mek[i], markersize=markersize, markeredgewidth=2) + # ax[3].plot(d_size[i], lose_chirps[i], 'o', color=lose_colors[i], markeredgecolor=mek[i], markersize=markersize, markeredgewidth=2 ) + # + # ax[0].set_ylabel('win chirps [n]', fontsize=12) + # ax[1].set_ylabel('lose chirps [n]', fontsize=12) + # ax[1].set_xlabel(u'$\Delta$size [cm]', fontsize=12) + # ax[3].set_xlabel(u'$\Delta$size [cm]', fontsize=12) + # + # plt.setp(ax[0].get_xticklabels(), visible=False) + # plt.setp(ax[2].get_xticklabels(), visible=False) + # + # plt.setp(ax[2].get_yticklabels(), visible=False) + # plt.setp(ax[3].get_yticklabels(), visible=False) + # plt.tick_params(labelsize=10) + # + # ### EODf + # ################################################################################################################## + # + # fig = plt.figure(figsize=(20/2.54, 12/2.54)) + # gs = gridspec.GridSpec(1, 1, left=0.1, bottom=0.15, right=0.95, top=0.95) + # ax = fig.add_subplot(gs[0, 0]) + # + # for i in range(len(lose_rises)): + # ax.plot(d_EODf[i], lose_rises[i], 'p', color=win_colors[i], markeredgecolor=mek[i], markersize=markersize) + # # ax.plot(d_EODf[i]*-1, lose_rises[i], 'o', color=lose_colors[i], markeredgecolor=mek[i], markersize=markersize) + # + # ax.set_ylabel('lose rises [n]', fontsize=12) + # ax.set_xlabel(u'$\Delta$EODf [Hz]', fontsize=12) + # plt.tick_params(labelsize=10) + # + # ################################################################################################################## + # fig = plt.figure(figsize=(20/2.54, 12/2.54)) + # gs = gridspec.GridSpec(1, 1, left=0.1, bottom=0.15, right=0.95, top=0.95) + # ax = fig.add_subplot(gs[0, 0]) + # + # for i in range(len(lose_chirps)): + # ax.plot(d_EODf[i], lose_chirps[i], 'p', color=win_colors[i], markeredgecolor=mek[i], markersize=markersize) + # # ax.plot(d_EODf[i]*-1, lose_chirps[i], 'o', color=lose_colors[i], markeredgecolor=mek[i], markersize=markersize) + # + # ax.set_ylabel('lose chirps [n]', fontsize=12) + # ax.set_xlabel(u'$\Delta$EODf [Hz]', fontsize=12) + # plt.tick_params(labelsize=10) def main(): trial_summary = pd.read_csv('trial_summary.csv', index_col=0) chirp_notes = pd.read_csv('chirp_notes.csv', index_col=0) - trial_summary = trial_summary[chirp_notes['good'] == 1] + # trial_summary = trial_summary[chirp_notes['good'] == 1] plot_rise_vs_chirp_count(trial_summary) - plot_chirp_rise_count_per_pairing(trial_summary) - - plot_chirp_rise_count_vs_size_diff(trial_summary) - + plot_chirp_rise_count_per_pairing(trial_summary, + beh_key_win='chirps_win', beh_key_lose='chirps_lose', + ylabel='chirps [n]') + plot_chirp_rise_count_per_pairing(trial_summary, + beh_key_win='rises_win', beh_key_lose='rises_lose', + ylabel='rises [n]') + + + plot_beh_count_vs_meta(trial_summary, + beh_key_win='chirps_win', beh_key_lose='chirps_lose', + meta_key_win="size_win", meta_key_lose='size_lose', + xlabel=u'$\Delta$size [cm]') + plot_beh_count_vs_meta(trial_summary, + beh_key_win='rises_win', beh_key_lose='rises_lose', + meta_key_win="size_win", meta_key_lose='size_lose', + xlabel=u'$\Delta$size [cm]') + plot_beh_count_vs_meta(trial_summary, + beh_key_win='chirps_win', beh_key_lose='chirps_lose', + meta_key_win="EODf_win", meta_key_lose='EODf_lose', + xlabel=u'$\Delta$EODf [Hz]') + plot_beh_count_vs_meta(trial_summary, + beh_key_win='rises_win', beh_key_lose='rises_lose', + meta_key_win="EODf_win", meta_key_lose='EODf_lose', + xlabel=u'$\Delta$EODf [Hz]') plt.show() - embed() - quit() + if __name__ == '__main__': main() \ No newline at end of file