square the number in python import math code example

Example: python square root

import math
a = input("what do you wnat to square root")
print(math.sqrt(a))