結果 : how to write 3d array in python