| 1 | #!/usr/bin/env python |
|---|
| 2 | |
|---|
| 3 | import os |
|---|
| 4 | import sys |
|---|
| 5 | import math |
|---|
| 6 | |
|---|
| 7 | ROOT = os.path.dirname(os.path.abspath(__file__)) |
|---|
| 8 | sys.path.insert(0, os.path.join(ROOT, '..')) |
|---|
| 9 | |
|---|
| 10 | from pygooglechart import ScatterChart |
|---|
| 11 | |
|---|
| 12 | import settings |
|---|
| 13 | import helper |
|---|
| 14 | |
|---|
| 15 | def scatter_random(): |
|---|
| 16 | chart = ScatterChart(settings.width, settings.height, |
|---|
| 17 | x_range=(0, 100), y_range=(0, 100)) |
|---|
| 18 | chart.add_data(helper.random_data()) |
|---|
| 19 | chart.add_data(helper.random_data()) |
|---|
| 20 | chart.download('scatter-random.png') |
|---|
| 21 | |
|---|
| 22 | def scatter_random_marker_sizes(): |
|---|
| 23 | chart = ScatterChart(settings.width, settings.height, |
|---|
| 24 | x_range=(0, 100), y_range=(0, 100)) |
|---|
| 25 | chart.add_data(helper.random_data()) |
|---|
| 26 | chart.add_data(helper.random_data()) |
|---|
| 27 | chart.add_data(helper.random_data()) |
|---|
| 28 | chart.download('scatter-random-marker-sizes.png') |
|---|
| 29 | |
|---|
| 30 | def scatter_circle(): |
|---|
| 31 | chart = ScatterChart(settings.width, settings.height, |
|---|
| 32 | x_range=(0, 100), y_range=(0, 100)) |
|---|
| 33 | steps = 40 |
|---|
| 34 | xradius = 25 |
|---|
| 35 | yradius = 45 |
|---|
| 36 | xmid = 50 |
|---|
| 37 | ymid = 50 |
|---|
| 38 | xlist = [] |
|---|
| 39 | ylist = [] |
|---|
| 40 | for angle in xrange(0, steps + 1): |
|---|
| 41 | angle = float(angle) / steps * math.pi * 2 |
|---|
| 42 | xlist.append(math.cos(angle) * xradius + xmid) |
|---|
| 43 | ylist.append(math.sin(angle) * yradius + ymid) |
|---|
| 44 | chart.add_data(xlist) |
|---|
| 45 | chart.add_data(ylist) |
|---|
| 46 | chart.add_data(range(len(ylist))) |
|---|
| 47 | chart.add_marker(0, 1.0, 'o', '00ff00', 10) |
|---|
| 48 | chart.download('scatter-circle.png') |
|---|
| 49 | |
|---|
| 50 | def main(): |
|---|
| 51 | scatter_random() |
|---|
| 52 | scatter_random_marker_sizes() |
|---|
| 53 | scatter_circle() |
|---|
| 54 | |
|---|
| 55 | if __name__ == '__main__': |
|---|
| 56 | main() |
|---|