Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Python | How to append elements to a list randomly

Tags:

python

Is there a way to append elements to a list randomly, built in function

ex:

def random_append():
     lst = ['a']
     lst.append('b')
     lst.append('c')
     lst.append('d')
     lst.append('e')
     return print lst

this will out put ['a', 'b', 'c', 'd', 'e']

But I want it to add elements randomly and out put something like this: ['b', 'd', 'b', 'e', 'c']

And yes there's a function random.shuffle() but it shuffles a list at once which I don't require, I just want to perform random inserts only.

like image 503
Switch Avatar asked Mar 19 '10 07:03

Switch


3 Answers

If you need to perform single insert in a random position then the already given trivial exapmle works:

from random import randrange, sample

def random_insert(lst, item):
    lst.insert(randrange(len(lst)+1), item)

However if you need to insert k items to a list of length n then using the previously given function is O(n*k + k**2) complexity. However inserting multiple items can be done in linear time O(n+k) if you calculate the target positions ahead of time and rewrite the input list in one go:

def random_insert_seq(lst, seq):
    insert_locations = sample(xrange(len(lst) + len(seq)), len(seq))
    inserts = dict(zip(insert_locations, seq))
    input = iter(lst)
    lst[:] = [inserts[pos] if pos in inserts else next(input)
        for pos in xrange(len(lst) + len(seq))]
like image 197
Ants Aasma Avatar answered Sep 20 '22 19:09

Ants Aasma


If there is supposed to be exactly one of each item

>>> from random import randint
>>> a=[]
>>> for x in "abcde":
...  a.insert(randint(0,len(a)),x)
... 
>>> a
['b', 'a', 'd', 'c', 'e']

If you are allowing duplicates (as the output indicates)

>>> from random import choice
>>> a=[choice("abcde") for x in range(5)]
>>> a
['a', 'b', 'd', 'b', 'a']
like image 22
John La Rooy Avatar answered Sep 19 '22 19:09

John La Rooy


random.shuffle is probably the best tool for the job. It is simple, obvious, and well-named—it's probably more readable than the other suggestions you will get. Additionally, using it is O(n), but using insert (an O(n) operation) n times is quadratic.

like image 37
Mike Graham Avatar answered Sep 16 '22 19:09

Mike Graham