From 38b4b6800c02c499d6ce7777fe7bccf14beb4666 Mon Sep 17 00:00:00 2001 From: Till Raab Date: Wed, 17 May 2023 15:15:15 +0200 Subject: [PATCH] sex assignment works. maybe rewrite trial analyis in a way that everything is saved with reference to ids not winner/loser... seems to be fishy. --- complete_analysis.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/complete_analysis.py b/complete_analysis.py index befefe6..b580ead 100644 --- a/complete_analysis.py +++ b/complete_analysis.py @@ -364,6 +364,21 @@ def main(data_folder=None): embed() quit() + + + for g in pd.unique(trial_summary['group']): + fish_no = np.unique(np.concatenate((trial_summary['win_fish'][trial_summary['group'] == g], + trial_summary['lose_fish'][trial_summary['group'] == g]))) + for f in fish_no: + fish_EODf25 = np.concatenate((trial_summary['EODf_lose'][(trial_summary['group'] == g) & (trial_summary['lose_fish'] == f)], + trial_summary['EODf_win'][(trial_summary['group'] == g) & (trial_summary['win_fish'] == f)])) + if np.median(fish_EODf25) < 740: + sex = 'f' + else: + 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 + pass if __name__ == '__main__':