why this error in forming array?
have you run this line import numpy as np
see here i am doing the same thing but i am getting the answers.