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