temp.py

# Define CSV column names
Rewards= {'collision_rew': [-0.26, -0.4, -0.61, -0.6, -0.56, -0.31, -0.45, -0.31, -0.16, -0.18, -0.23, -0.28, -0.23, -0.18, -0.18, -0.27, -0.11, -0.16, -0.19, -0.1, -0.04], 
          'right_lane': [12.48, 11.62, 10.72, 10.02, 12.11, 10.17, 12.35, 13.69, 15.03, 9.72, 13.88, 8.89, 12.7, 12.16, 13.25, 11.88, 11.54, 14.9, 14.75, 13.87, 14.81], 
          'right_lane_rew': [2.4959999999999996, 2.3240000000000007, 2.144, 2.004, 2.4219999999999997, 2.0340000000000003, 2.469999999999999, 2.7380000000000013, 3.0060000000000002, 1.944, 2.775999999999999, 1.777999999999999, 2.5399999999999996, 2.4319999999999995, 2.65, 2.3759999999999994, 2.3080000000000003, 2.9799999999999986, 2.9500000000000006, 2.7740000000000014, 2.9620000000000006], 
          'speed_rew': [3.2998973059361494, 5.82687474862131, 7.196766897513154, 8.065869544328512, 8.110105740300114, 8.94078638917179, 8.416769828711837, 8.97014660873796, 8.785703998114535, 9.957393751098905, 8.712780047785984, 9.80619601905509, 9.356197853411757, 9.85840751398171, 9.596933628960665, 9.61365767149695, 10.299340249358757, 9.362779329451913, 9.87360565325151, 9.681064556184042, 9.814429250623062],
          'tailgating_rew': [-0.44, -0.46399999999999975, -0.5199999999999997, -0.6399999999999998, -0.5039999999999998, -0.2320000000000001, -0.5519999999999998, -0.2640000000000001, -0.24000000000000007, -0.2960000000000001, -0.20800000000000005, -0.21600000000000005, -0.52, -0.14400000000000002, -0.19200000000000003, -0.2800000000000001, -0.2720000000000001, -0.12800000000000003, -0.3280000000000001, -0.16799999999999998, -0.04], 
          'tailgating': [0.55, 0.58, 0.65, 0.8, 0.63, 0.29, 0.69, 0.33, 0.3, 0.37, 0.26, 0.27, 0.65, 0.18, 0.24, 0.35, 0.34, 0.16, 0.41, 0.21': [0.6, 2.48, 2.2, 1.24, 0, 0.38, 1.48, 0.68, 0.69], 'true_reward': [21.96676914954342, 21.478546717888317, 19.700729310945725, 21.232918465205312, 22.4193994210209, 25.61190815989904, 23.341397809611557, 25.74442496379455, 28.341314998821584, 26.182537761103486, 26.231904196532906, 25.955455845242774, 26.08995699171567, 27.215254696238556, 27.34025018476708, 25.68303604468559, 28.497504322515905, 28.03007041424077, 28.063086866615507, 28.527165347615032, 29.912934948306077], 'feedback': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]}

columns = ['collision_rew', 'right_lane', 'right_lane_rew', 'speed_rew', 'tailgating_rew', 'tailgating', 'human_rew', 'lane_changed', 'true_reward', 'seed', 'iter']

# Open a new CSV file for writing
with open('rewards_data.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(columns)  # Write the column headers
    
    # Write data rows
    for i in range(len(data['collision_rew'])):
        row = [
            data['collision_rew'][i],
            data['right_lane'][i],
            data['right_lane_rew'][i],
            data['speed_rew'][i],
            data['tailgating_rew'][i],
            data['tailgating'][i],
            human_rew[i],
            lane_changed[i],
            data['true_reward'][i],
            seed[i],
            iter_[i]
        ]
        writer.writerow(row)

print("CSV file has been created successfully.")