[python] Find length of 2D array Python