結果 : how to get sub array in python