Python Reshape Array Toilet