generated from mwc/lab_scatter
	scatterplot.py
This commit is contained in:
		@@ -16,6 +16,7 @@ from plotting import (
 | 
				
			|||||||
    draw_y_tick,
 | 
					    draw_y_tick,
 | 
				
			||||||
    draw_point, 
 | 
					    draw_point, 
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
from transform import (
 | 
					from transform import (
 | 
				
			||||||
    maximum, 
 | 
					    maximum, 
 | 
				
			||||||
    minimum,
 | 
					    minimum,
 | 
				
			||||||
@@ -28,19 +29,42 @@ from transform import (
 | 
				
			|||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def draw_scatterplot(data, size=5, color="black"):
 | 
					def draw_scatterplot(data, size=5, color="black"):
 | 
				
			||||||
    "Draws a scatter plot, showing the data"
 | 
					 | 
				
			||||||
    prepare_screen()
 | 
					    prepare_screen()
 | 
				
			||||||
    draw_axes(data)
 | 
					    draw_axes(data)
 | 
				
			||||||
    draw_points(data, color, size)
 | 
					    draw_points(data, color, size)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def draw_axes(data):
 | 
					def draw_axes(data):
 | 
				
			||||||
    "Draws the scatter plot's axes."
 | 
					    draw_x_axis()
 | 
				
			||||||
 | 
					    x_values = get_x_values(data)
 | 
				
			||||||
def draw_points(data, color, size):
 | 
					    xmin, xmax = bounds(x_values)
 | 
				
			||||||
    "Draws the scatter plot's points."
 | 
					    ticks = get_tick_values(xmin, xmax)
 | 
				
			||||||
 | 
					    for tick in ticks:
 | 
				
			||||||
 | 
					        screen_x_position = scale(tick, xmin, xmax, 0, constants.PLOT_WIDTH)
 | 
				
			||||||
 | 
					        draw_x_tick(screen_x_position, tick)
 | 
				
			||||||
 | 
					   
 | 
				
			||||||
 | 
					    draw_y_axis()
 | 
				
			||||||
 | 
					    y_values = get_y_values(data)
 | 
				
			||||||
 | 
					    ymin, ymax = bounds(y_values)
 | 
				
			||||||
 | 
					    ticks = get_tick_values(ymin, ymax)
 | 
				
			||||||
 | 
					    for tick in ticks:
 | 
				
			||||||
 | 
					        screen_y_position = scale(tick, ymin, ymax, 0, constants.PLOT_HEIGHT)
 | 
				
			||||||
 | 
					        draw_y_tick(screen_y_position, tick)
 | 
				
			||||||
 | 
					    
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					def draw_points(data, color, size):    
 | 
				
			||||||
 | 
					    x_values = get_x_values(data)
 | 
				
			||||||
 | 
					    xmin, xmax = bounds(x_values)
 | 
				
			||||||
 | 
					    y_values = get_y_values(data)
 | 
				
			||||||
 | 
					    ymin, ymax = bounds(y_values)
 | 
				
			||||||
 | 
					    for x, y in data: # the loop values must match the 
 | 
				
			||||||
 | 
					        # draw_point function parameters in the plotting.py file
 | 
				
			||||||
 | 
					        scaled_x=scale(x, xmin, xmax, 0, constants.PLOT_WIDTH)
 | 
				
			||||||
 | 
					        scaled_y=scale(y, ymin, ymax, 0, constants.PLOT_HEIGHT) 
 | 
				
			||||||
 | 
					        draw_point(x, y, color, size)
 | 
				
			||||||
 | 
					   
 | 
				
			||||||
with no_delay():
 | 
					with no_delay():
 | 
				
			||||||
    data = generate_data(50, 10, 500, 5, 400, 1000)
 | 
					    data = generate_data(50, 10, 500, 5, 400, 1000)
 | 
				
			||||||
    draw_scatterplot(data, size=5, color="blue")
 | 
					    draw_scatterplot(data, size=5, color="blue")
 | 
				
			||||||
    hideturtle()
 | 
					    hideturtle()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
done()
 | 
					done()
 | 
				
			||||||
 
 | 
				
			|||||||
@@ -15,7 +15,7 @@ def test(function, arguments, expected):
 | 
				
			|||||||
    if observed != expected:
 | 
					    if observed != expected:
 | 
				
			||||||
       args = ', '.join(str(arg) for arg in arguments) 
 | 
					       args = ', '.join(str(arg) for arg in arguments) 
 | 
				
			||||||
       print(f"Error: Expected {function}({args}) to equal {expected}, but it was {observed}")
 | 
					       print(f"Error: Expected {function}({args}) to equal {expected}, but it was {observed}")
 | 
				
			||||||
 | 
					    
 | 
				
			||||||
 | 
					
 | 
				
			||||||
test(maximum, [[0, 1, 2, 3]], 3)
 | 
					test(maximum, [[0, 1, 2, 3]], 3)
 | 
				
			||||||
test(maximum, [[-10, -20, -30]], -10)
 | 
					test(maximum, [[-10, -20, -30]], -10)
 | 
				
			||||||
@@ -33,4 +33,5 @@ test(ratio, [4, 10, 20], 0.0)
 | 
				
			|||||||
test(scale, [4, 0, 10, 0, 100], 40)
 | 
					test(scale, [4, 0, 10, 0, 100], 40)
 | 
				
			||||||
test(scale, [180, 120, 240, 0, 100], 50)
 | 
					test(scale, [180, 120, 240, 0, 100], 50)
 | 
				
			||||||
test(get_x_values, [[[0, 5], [1, 5], [2, 5]]], [0, 1, 2])
 | 
					test(get_x_values, [[[0, 5], [1, 5], [2, 5]]], [0, 1, 2])
 | 
				
			||||||
test(get_y_values, [[[0, 5], [1, 5], [2, 5]]], [5, 5, 5])
 | 
					test(get_y_values, [[[0, 5], [1, 5], [2, 5]]], [5, 5, 5])
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										20
									
								
								transform.py
									
									
									
									
									
								
							
							
						
						
									
										20
									
								
								transform.py
									
									
									
									
									
								
							@@ -4,7 +4,6 @@
 | 
				
			|||||||
# The functions in this module transform data. 
 | 
					# The functions in this module transform data. 
 | 
				
			||||||
# None of them are finished; this is your job!
 | 
					# None of them are finished; this is your job!
 | 
				
			||||||
 | 
					
 | 
				
			||||||
import sys
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
def maximum(data):
 | 
					def maximum(data):
 | 
				
			||||||
    highest = None
 | 
					    highest = None
 | 
				
			||||||
@@ -49,8 +48,8 @@ def clamp(value, low, high):
 | 
				
			|||||||
        return high
 | 
					        return high
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def ratio(value, start, end):
 | 
					def ratio(value, domain_min, domain_max):
 | 
				
			||||||
    ratio = (value - start)/(end - start)
 | 
					    ratio = (value - domain_min)/(domain_max - domain_min)
 | 
				
			||||||
    return clamp(ratio,0,1)
 | 
					    return clamp(ratio,0,1)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -61,12 +60,15 @@ def scale(value, domain_min, domain_max, range_min, range_max):
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
def get_x_values(points):
 | 
					def get_x_values(points):
 | 
				
			||||||
    for (x, y) in points:
 | 
					    x_values=[]
 | 
				
			||||||
        return ([x], )
 | 
					    for ([x, y]) in points:
 | 
				
			||||||
 | 
					        x_values.append(x)
 | 
				
			||||||
 | 
					    return x_values
 | 
				
			||||||
    
 | 
					    
 | 
				
			||||||
def get_y_values(points):
 | 
					def get_y_values(points):
 | 
				
			||||||
    for (x, y) in points:
 | 
					    y_values=[]
 | 
				
			||||||
        return ([y], )
 | 
					    for ([x, y]) in points:
 | 
				
			||||||
       
 | 
					        y_values.append(y)
 | 
				
			||||||
    
 | 
					    return y_values
 | 
				
			||||||
 | 
					  
 | 
				
			||||||
        
 | 
					        
 | 
				
			||||||
		Reference in New Issue
	
	Block a user