# Simulate a simple physics problem time = np.linspace(0, 10, 1000) # Time from 0 to 10 seconds gravity = 9.81 # m/s² initial_velocity = 50 # m/s # Calculate position (physics equation) position = initial_velocity * time - 0.5 * gravity * time**2 # Find maximum height max_height = np.max(position) max_time = time[np.argmax(position)] print(f"Maximum height: {max_height:.1f}m at {max_time:.1f}s")