Python Return Array Dimensions