scatterplot.py

This commit is contained in:
njmason2
2025-10-12 18:24:23 -04:00
parent 1c4cee2704
commit 640ff70865
3 changed files with 43 additions and 16 deletions

View File

@@ -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)
xmin, xmax = bounds(x_values)
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): def draw_points(data, color, size):
"Draws the scatter plot's points." 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()

View File

@@ -34,3 +34,4 @@ 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])

View File

@@ -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