python给数组每个元素赋值