csv-bivariate-normal-distribution: added

tikz/csv-bivariate-normal-distribution/create.py

@@ -0,0 +1,34 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""Create samples for bivariate distribution"""
+
+from numpy.random import multivariate_normal, seed
+
+
+def create_data(n):
+    means = [0.0, 0.0]
+    cov = [[1.0, 0.7], [0.7, 4.0]]
+    seed(0)
+    samples = multivariate_normal(means, cov, n)
+    with open("data.csv", "w") as f:
+        f.write("x,y\n")
+        for datapoint in samples:
+            f.write("%0.4f,%0.4f\n" % tuple(datapoint))
+
+
+def get_parser():
+    from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
+    parser = ArgumentParser(description=__doc__,
+                            formatter_class=ArgumentDefaultsHelpFormatter)
+    parser.add_argument("-n",
+                        dest="n",
+                        default=5000,
+                        type=int,
+                        help="Number of points to generate")
+    return parser
+
+
+if __name__ == "__main__":
+    args = get_parser().parse_args()
+    create_data(args.n)