# -*- coding: utf-8 -*-
from __future__ import division
import numpy as np
import matplotlib.pyplot as plt

#
red = "#e41a1c"
gray = "#666666"
plt.rcParams["mathtext.fontset"] = "cm"
plt.rcParams["axes.spines.top"] = False
plt.rcParams["axes.spines.right"] = False
plt.rcParams["axes.edgecolor"] = gray
plt.rcParams["xtick.color"] = gray
plt.rcParams["ytick.color"] = gray


a, b, c, d = 1, 1, 1, 1
y, x, = np.mgrid[-1:1:101j, -1:1:101j]
z = a*x**3 + b*x**2*y + c*x*y**2 + x*y - d*y**3

fig, ax = plt.subplots(figsize=(4, 3))
plt.contour(x, y, z, [0], colors=red)
plt.xlabel(r"$x$", fontsize=16)
plt.ylabel(r"$y$", rotation="horizontal", fontsize=16)
ax.spines['bottom'].set_position(('data', 0))
ax.spines['left'].set_position(('data', 0))
plt.xticks([-1, -0.5, 0.5, 1])
plt.yticks([-1, -0.5, 0.5, 1])
ax.xaxis.set_label_coords(1.0, 0.6)
ax.yaxis.set_label_coords(0.55, 0.95)
plt.grid(alpha=0.2)
plt.savefig("newton_trident_y=inf.svg")
plt.show()