Two Dimensional Array In Python Numpy Functions