:orphan:

grm/plot
--------


[GRM] Contourf and series legend example


.. code-block:: python
    
    
    import math
    import sys
    import numpy as np
    import grm
    from random import random
    
    
    def test_consecutive_plots():
        n = 1000
        x_vals = np.linspace(0, 2 * math.pi, n)
        plots = [[x_vals, 2 * np.sin(x_vals)], [x_vals, np.sin(x_vals)]]
    
        print("filling argument container...")
    
        args = grm.args.new()
        for i in range(0, 2):
            args["x"] = plots[i][0]
            args["y"] = plots[i][1]
            grm.plot.plot(args)
            print("Press any key to continue...")
            sys.stdin.read(1)
    
    
    def test_line():
        n = 1000
        x_vals = np.linspace(0, 2 * math.pi, n)
    
        print("filling argument container...")
    
        args = grm.args.new()
        args["series"] = [{"x": x_vals, "y": np.sin(x_vals)}, {"x": x_vals, "y": np.cos(x_vals)}]
        args["labels"] = ["sin", "cos"]
        args["kind"] = "line"
    
        print("plotting data...")
    
        grm.plot.plot(args)
    
        print("Press any key to continue...")
        sys.stdin.read(1)
    
    
    def test_contourf():
        x = []
        y = []
    
        n = 100
    
        for i in range(0, n):
            x.append(random() * 8.0 - 4.0)
            y.append(random() * 8.0 - 4.0)
    
        z = np.sin(x) + np.cos(y)
    
        print("filling argument container...")
    
        args = grm.args.new({"subplots": {"series": {"x": x, "y": y, "z": z}, "kind": "contourf"}})
    
        print("plotting data...")
    
        grm.plot.plot(args)
    
        print("Press any key to continue...")
        sys.stdin.read(1)
    
    
    if __name__ == "__main__":
        test_line()
        test_consecutive_plots()
        test_contourf()
        grm.plot.finalize()