mandelbrot start

JavaProgrammerLB · JavaProgrammerLB · commit 1146a94ef57e · 2018-07-10T16:18:45.000+08:00
diff --git a/ufo/mandelbrot0.py b/ufo/mandelbrot0.py
@@ -0,0 +1,26 @@
+import numpy as np
+import matplotlib.pyplot as plt
+
+
+def main():
+    plt.imshow(mandelbrot(400, 400))
+    plt.show()
+
+
+def mandelbrot(w, h, maxit=200):
+    y, x = np.ogrid[-1.4:1.4:h * 1j, -2:0.8:w * 1j]
+    c = x + y * 1j
+    z = c
+    divtime = maxit + np.zeros(z.shape, dtype=int)
+
+    for i in range(maxit):
+        z = z ** 2 + c
+        diverge = z * np.conj(z) > 2 ** 2
+        div_now = diverge & (divtime == maxit)
+        divtime[div_now] = i
+        z[diverge] = 2
+    return divtime
+
+
+if __name__ == "__main__":
+    main()
diff --git a/ufo/numpyogrid.py b/ufo/numpyogrid.py
@@ -0,0 +1,20 @@
+import numpy as np
+
+
+def main():
+    first_ogid()
+
+
+def first_ogid():
+    x, y = np.ogrid[1:4:1, 1:5:2]
+    print(x)
+    print(y)
+    x, y = np.ogrid[1:4:3j, 1:5:2j]
+    print(x)
+    print(y)
+    x=np.ogrid[1:5:2]
+    print(x)
+
+
+if __name__ == "__main__":
+    main()
