Help with homework

Question

I want the solution from the equation (solve1) to be the 'y' axis. How can I do this?? It works if I print the solution from the equation and then paste it instead of solve1. But I cant do it directly. Please help me with this.

import sympy as sp
import matplotlib.pyplot as plt
import numpy as np
from scipy import constants as constant

h1 = sp.Symbol('h1')
v1 = sp.Symbol('v1')

h0 = 2.0
v0 = 20
a = 30
g = constant.g
m = 1

eqn = sp.Eq(m*g*h0 + 0.5*m*v0**2, m*g*h1 + 0.5*m*v1**2)
solve = (sp.solve(eqn))
solve1 = solve[0][h1]

v1 = np.linspace(1,10,10)
y = solve1

print(y)
plt.plot(v1,y)

Answer 1

https://code.sololearn.com/cfsADQ1rFLbf/?ref=app

Ugly but it works