Heart shape in Python (matplotlib)

heart.py

import numpy as np
import matplotlib.pyplot as plt

# Define the implicit function for the heart shape
def heart_shape(x, y):
    return (x**2 + y**2 - 1)**3 - x**2 * y**3

# Create a grid of x, y points
x = np.linspace(-1.5, 1.5, 400)
y = np.linspace(-1.5, 1.5, 400)
x, y = np.meshgrid(x, y)

# Compute the function values
z = heart_shape(x, y)

# Plotting
plt.figure(figsize=(6, 6))
plt.contour(x, y, z, levels=[0], colors='red')
plt.title('Heart Shape Plot')
plt.axis('equal')
plt.show()