結果 : sub array problems in python