diff --git a/scatterplot.py b/scatterplot.py index 50f66fb..90eb1dc 100644 --- a/scatterplot.py +++ b/scatterplot.py @@ -34,13 +34,50 @@ def draw_scatterplot(data, size=5, color="black"): draw_points(data, color, size) def draw_axes(data): - "Draws the scatter plot's axes." + + draw_x_axis() + x_values = get_x_values(data) + xmin, xmax = bounds(x_values) + x_ticks = get_tick_values(xmin, xmax) + + for tick in x_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) + y_ticks = get_tick_values(ymin, ymax) + + for tick in y_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." + x_values = get_x_values(data) + y_values= get_y_values(data) + xmin, xmax = bounds(x_values) + ymin, ymax = bounds(y_values) + + for x,y in data: + x = clamp(x, xmin, xmax) + ratio(x, xmin, xmax) + scaled_x=scale(x, xmin, xmax, 0, constants.PLOT_WIDTH) + + y = clamp(y, ymin, ymax) + ratio(y, ymin, ymax) + scaled_y=scale(y, ymin, ymax, 0, constants.PLOT_HEIGHT) + #At this point I am deeply confused with what domain min and max are supposed to be. + # I am also sort of confused of how much of the individual recalling is needed for the functions such as ratio, clamp, etc. + draw_point(scaled_x, scaled_y, color, size) + + with no_delay(): data = generate_data(50, 10, 500, 5, 400, 1000) draw_scatterplot(data, size=5, color="blue") hideturtle() done() + +