python将for循环结果写成数组