結果 : how to partition an array in python