generated from mwc/lab_scatter
143 lines
3.7 KiB
Python
143 lines
3.7 KiB
Python
# scatterplot.py
|
|
# ------------
|
|
# By MWC Contributors
|
|
# Uses lots of helper functions in other modules to draw a scatter plot.
|
|
|
|
from math import floor, ceil, log
|
|
|
|
from turtle import *
|
|
from superturtle.movement import no_delay
|
|
import constants
|
|
|
|
from generate_data import generate_data
|
|
from ticks import get_tick_values
|
|
from plotting import (
|
|
prepare_screen,
|
|
draw_x_axis,
|
|
draw_y_axis,
|
|
draw_x_tick,
|
|
draw_y_tick,
|
|
draw_point,
|
|
)
|
|
from superturtle.movement import no_delay
|
|
import constants
|
|
def flyto(x, y):
|
|
penup()
|
|
goto(x, y)
|
|
pendown()
|
|
from generate_data import generate_data
|
|
from ticks import get_tick_values
|
|
from plotting import (
|
|
prepare_screen,
|
|
draw_point,
|
|
)
|
|
from transform import (
|
|
bounds,
|
|
scale,
|
|
get_x_values,
|
|
get_y_values
|
|
|
|
)
|
|
|
|
def draw_x_axis(min_val, max_val):
|
|
"Draws the x-axis of the scatter plot from min_val to max_val"
|
|
penup
|
|
flyto(0,0)
|
|
goto(constants.PLOT_WIDTH,0)
|
|
pendown()
|
|
|
|
def get_tick_values(low, high):
|
|
"""Returns a list of values to use for ticks (labeled points along an axis).
|
|
Includes the lowest value, a bunch of "nice" intermediate values, and the highest value.
|
|
"""
|
|
tick_interval = get_tick_interval(high - low)
|
|
first_tick = ceil(low / tick_interval) * tick_interval
|
|
return [low] + list(range(first_tick, high, tick_interval)) + [high]
|
|
|
|
def get_tick_interval(span):
|
|
"""Returns a 'nice' interval for ticks across span.
|
|
The interval is a power of ten (e.g. 1000, 100, 0.1)
|
|
scaled so that there will be between 0 and 10 internal ticks.
|
|
"""
|
|
log_span = log(span, 10)
|
|
return 10 ** floor(log_span)
|
|
|
|
|
|
def draw_y_axis(min_val, max_val):
|
|
"Draws the y-axis of the scatterplot from min_val to max_val"
|
|
penup()
|
|
flyto(0,0)
|
|
goto(0,constants.PLOT_HEIGHT)
|
|
pendown
|
|
|
|
|
|
|
|
|
|
def draw_x_tick(position, label):
|
|
"Draws a tick mark on the x-axis at the specified position"
|
|
flyto(position, 0)
|
|
goto(position, -constants.TICK_LENGTH)
|
|
flyto(position, -constants.TICK_LENGTH - 10)
|
|
write(label, align='center')
|
|
|
|
|
|
|
|
def draw_y_tick(position, label):
|
|
"Draws a tick mark on the y-axis at the specified position"
|
|
flyto(0, position)
|
|
goto(-constants.TICK_LENGTH, position)
|
|
write(label, align='right')
|
|
|
|
|
|
def get_tick_values(low, high):
|
|
"""Returns a list of values to use for ticks (labeled points along an axis).
|
|
Includes the lowest value, a bunch of "nice" intermediate values, and the highest value.
|
|
"""
|
|
tick_interval = get_tick_interval(high - low)
|
|
first_tick = ceil(low / tick_interval) * tick_interval
|
|
return [low] + list(range(first_tick, high, tick_interval)) + [high]
|
|
|
|
|
|
|
|
def draw_scatterplot(data, size=5, color="black"):
|
|
"Draws a scatter plot, showing the data"
|
|
prepare_screen()
|
|
draw_axes(data)
|
|
draw_points(data, size, color)
|
|
|
|
def draw_axes(data):
|
|
"Draws the scatter plot's axes."
|
|
x_values = get_x_values(data)
|
|
y_values = get_y_values(data)
|
|
xmin, xmax = bounds(x_values)
|
|
ymin, ymax = bounds(y_values)
|
|
draw_x_axis(xmin, xmax)
|
|
draw_y_axis(ymin, ymax)
|
|
|
|
|
|
|
|
def draw_points(x,y, color="blue", size=5):
|
|
"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 point in data:
|
|
x, y = point
|
|
|
|
screen_x = scale(x, xmin, xmax, 0, constants.PLOT_WIDTH)
|
|
screen_y = scale(y, ymin, ymax, 0, constants.PLOT_HEIGHT)
|
|
|
|
|
|
draw_point(screen_x, screen_y, color, size)
|
|
|
|
|
|
|
|
with no_delay():
|
|
data = generate_data(50, 10, 500, 5, 400, 1000)
|
|
draw_scatterplot(data, size=5, color="blue")
|
|
hideturtle()
|
|
done()
|
|
|