Map Function For Array Python