from sage.all import *


import csv
file='Copy of Data Challenge 1 1.csv'
reader=csv.reader(open(file))
all_data=[]
for row in reader:
    all_data.append(row)
    
data=list([all_data])

x_sections = [0, 5, 10, 15, 20]
y_sections = [0, 5, 10]

sections = [[[] for _ in range(len(y_sections)-1)] for _ in range(len(x_sections)-1)]

for i in range(len(x_sections)-1):
    for j in range(len(y_sections)-1):
        x_min = x_sections[i]
        x_max = x_sections[i+1]
        y_min = y_sections[j]
        y_max = y_sections[j+1]
        
        for point in all_data:
            x = float(point[0])
            y = float(point[1])
            T = float(point[2])  
            
            if x_min <= x <= x_max and y_min <= y <= y_max:
                sections[i][j].append((x, y, T))
                

for i in range(len(x_sections)-1):
    for j in range(len(y_sections)-1):
        section_length = len(sections[i][j])
        print(f"Section ({x_sections[i]}-{x_sections[i+1]}, {y_sections[j]}-{y_sections[j+1]}): {sections[i][j]}, Length: {section_length}""\n")

section_averages = []
for i in range(len(x_sections)-1):
    for j in range(len(y_sections)-1):
        section = sections[i][j]
        z_values = [point[2] for point in section]  
        
        section_average = sum(z_values) / len(z_values)
        section_averages.append(section_average)
        print(f"Section ({x_sections[i]}-{x_sections[i+1]}, {y_sections[j]}-{y_sections[j+1]}), Average T: {section_average}""\n")

total_average = sum(section_averages) / len(section_averages)
print("Total Weighted Temp:", total_average)