visual_zeropole.py 412 B

1234567891011121314151617
  1. import matplotlib.pyplot as plt
  2. a = []
  3. for i in range(-500, 500, 1):
  4. row = []
  5. for j in range(-500, 500, 1):
  6. x, y = float(j), float(i)
  7. x /= 2000
  8. y /= 2000
  9. v = y**2 - x**3 - 5
  10. #if 0.98 < x < 1.02 and 2.4 < y < 2.6:
  11. # print(v)
  12. v = int(v * 1000)
  13. row.append(v)
  14. a.append(row)
  15. plt.imshow(a, cmap='hot', interpolation='nearest')
  16. plt.show()