Python Assign Array