Sí me di cuenta después de que lo envié y se me pasó jajj, ahí lo arregle,
Te podes definir cualquier función sí, la pones en la parte de f(x1,x2), vas cambiando los dxi en función del sistema que quieras plotear, otra cosa que se me pasó decir es que si le dan a shift+tab python te muestra todas las variables y una explicación de estas y de la función + ejemplos de uso.
Saludos