Create a Python script that takes an array with n elements and shuffles it.
n = 10
import random

def shuffle(lst):
    random.shuffle(lst)
    return lst

arr = [i for i in range(10)]
print(shuffle(arr))