Assign Value In Array Python