> For the complete documentation index, see [llms.txt](https://george-jen.gitbook.io/data-science-and-apache-spark/llms.txt). Markdown versions of documentation pages are available by appending `.md` to page URLs; this page is available as [Markdown](https://george-jen.gitbook.io/data-science-and-apache-spark/map-and-filter-takes-function-as-input.md).

# map and filter takes function as input

### map and filter takes function as input.

map and filter takes function as input, you can define one if it has more than couple lines using def, if the function is short and simple, typically use lambda function.

example:

```
#even element from list
def even(N):
    if N%2==0:
        return True
    else:
        return False

a=[1,2,3,4,5,6]
list(filter(even,a))
```

```
#square each list element
def square(N):
    return N*N
a=[1,2,3,4,5,6]
list(map(square,a))
```