def calFirstIntentionStepRationAfterNoise(noisePoints, goalList):
afterNoiseGoalList = goalList[noisePoints:]
afterNoiseFirstIntentionStep = calculateFirstIntentionStep(afterNoiseGoalList)
return afterNoiseFirstIntentionStep
from dataAnalysis import calculateSE, calculateFirstIntentionStep
if __name__ == '__main__':
resultsPath = os.path.join(os.path.join(DIRNAME, '..'), 'results')
statsList = []
stdList = []
# participants = ['human', 'softmaxBeta0.5', 'rewardVariance5','rewardVariance10', 'rewardVariance30' ,'rewardVariance50']
participants = ['human', 'softmaxBeta0.45', 'softmaxBeta0.5']
for participant in participants:
dataPath = os.path.join(resultsPath, participant)
df = pd.concat(map(pd.read_csv,
glob.glob(os.path.join(dataPath, '*.csv'))),
sort=False)
df['firstIntentionStep'] = df.apply(
lambda x: calculateFirstIntentionStep(eval(x['goal'])), axis=1)
df['totalStep'] = df.apply(lambda x: len(eval(x['trajectory'])),
axis=1)
# df.to_csv("all.csv")
nubOfSubj = len(df["name"].unique())
print(participant, nubOfSubj)
# dfExpTrail = df[(df['areaType'] == 'expRect') & (df['noiseNumber'] != 'special')]
# dfExpTrail = df[(df['distanceDiff'] == 0) & (df['areaType'] != 'none')]
# dfExpTrail = df[(df['distanceDiff'] == 0) & (df['areaType'] == 'midLine')]
# dfExpTrail = df[(df['distanceDiff'] == 0) & (df['areaType'] == 'straightLine')]
# dfExpTrail = df[(df['distanceDiff'] == 0) & (df['areaType'] == 'straightLine') & (df['intentionedDisToTargetMin'] == 2)]
# dfExpTrail = df[(df['areaType'] == 'straightLine') | (df['areaType'] == 'midLine') & (df['distanceDiff'] == 0)]
# dfExpTrail = df[(df['areaType'] != 'none')]
if __name__ == '__main__':
resultsPath = os.path.join(os.path.join(DIRNAME, '..'), 'results')
statsList = []
stdList = []
participants = ['human', 'softmaxBeta2.5']
for participant in participants:
dataPath = os.path.join(resultsPath, participant)
df = pd.concat(map(pd.read_csv, glob.glob(os.path.join(dataPath, '*.csv'))), sort=False)
df['avoidCommitmentZone'] = df.apply(lambda x: calculateAvoidCommitmnetZone(eval(x['playerGrid']), eval(x['target1']), eval(x['target2'])), axis=1)
df['avoidCommitmentRatio'] = df.apply(lambda x: calculateFirstOutZoneRatio(eval(x['trajectory']), x['avoidCommitmentZone']), axis=1)
df['firstIntentionStep'] = df.apply(lambda x: calculateFirstIntentionStep(eval(x['goal'])), axis=1)
df['firstIntentionRatio'] = df.apply(lambda x: calculateFirstIntentionRatio(eval(x['goal'])), axis=1)
# df.to_csv("all.csv")
nubOfSubj = len(df["name"].unique())
print(participant, nubOfSubj)
dfExpTrail = df[(df['areaType'] == 'expRect') & (df['noiseNumber'] != 'special')]
# dfExpTrail = df[(df['distanceDiff'] == 0) & (df['areaType'] != 'none')]
# dfExpTrail = df[(df['distanceDiff'] == 0) & (df['areaType'] == 'straightLine')]
# dfExpTrail = df[(df['distanceDiff'] == 0) & (df['areaType'] == 'midLine') & (df['intentionedDisToTargetMin'] > 2)]
# dfExpTrail = df[(df['distanceDiff'] == 0) & (df['areaType'] == 'midLine')]
# dfExpTrail = df[(df['distanceDiff'] == 0) & (df['areaType'] == 'midLine')]
# dfExpTrail = df[((df['distanceDiff'] == 0) & df['areaType'] == 'straightLine') | (df['areaType'] == 'midLine')]
