CODEWITHSUNDEEP
pythonlistrandomlist-comprehension
Seeven
Seeven

Reputation: 1079

Using Random in Python's list comprehension

I have a list of words named words and I would like to generate a list of 100 three-words elements named pwds. I want these 3 words to be randomly picked from the words list for each element of the pwds list, and I'd like to use list comprehension to do this.

This solution works :

pwds = [random.choice(words)+' '+random.choice(words)+' '+random.choice(words) for i in range(0,100)]

It generates a list that looks like : ['correct horse battery', 'staple peach peach', ...]

But I was looking for a way that prevents repetiting 3 times random.choice(words), so I tried this :

pwds = [(3*(random.choice(words)+' ')).strip() for i in range(0,100)]

But unfortunately, this solution makes each element having the same word three times (for example : ['horse horse horse', 'staple staple staple', ...]), which is expected to happend.

Do you know a way to pick 3 random words without repetition (EDIT : by "repetition", I mean the code repetition, not the random words repetition) ?

EDIT : My question is different than the one it has been marked as duplicate of because I'm looking for using list comprehension here. I know how I could generate different numbers, I'm just looking for specific way to do it.

Upvotes: 0

Views: 2655

Answers (2)

Gabz
Gabz

Reputation: 134

You can use the join function and a list comprehension to not repeat random.choice

pwds = [' '.join([random.choice(words) for _ in range(3)]) for _ in range(100)]

Upvotes: 1

jonrsharpe
jonrsharpe

Reputation: 122024

If you want the words to be able to repeat within each triplet, I think what you want is something like:

pwds = [" ".join(random.choice(words) for _ in range(3)) for _ in range(100)]

Note the use of _ to indicate that we don't actually use the numbers generated by either range, and the fact that range(0, n) is the same as range(n).

A somewhat shorter example:

>>> import random
>>> words = ['correct', 'horse', 'battery', 'staple']
>>> [" ".join(random.choice(words) for _ in range(3)) for _ in range(5)]
['horse horse correct', 
 'correct staple staple', 
 'correct horse horse', 
 'battery staple battery', 
 'horse battery battery']

Upvotes: 2

Related Questions