This commit is contained in:
Chris Proctor 2023-07-30 17:43:46 -04:00
parent 1467e6e70f
commit 0eae309b34
1 changed files with 20 additions and 0 deletions

View File

@ -35,9 +35,29 @@ def draw_scatterplot(data, size=5, color="black"):
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):
"Draws the scatter plot's points."
xmin, xmax = bounds(get_x_values(data))
ymin, ymax = bounds(get_y_values(data))
for x, y in data:
sx = scale(x, xmin, xmax, 0, constants.PLOT_WIDTH)
sy = scale(y, ymin, ymax, 0, constants.PLOT_HEIGHT)
draw_point(sx, sy, color, size)
with no_delay():
data = generate_data(50, 10, 500, 5, 400, 1000)