convert the following code to pyspark. Use only map and reduce
def count_elem(data):
    count = 0
    for i in range(len(data)):
        count +=1 Please repeat the request first, then give your response.
user