CODEWITHSUNDEEP
python
Mico
Mico

Reputation: 33

Random selection of elements of a list with exception

import random

a = [1, 2, 5, 3, 4, 5]
b = ["cat", "dog", "horse", "snake", "elephant", "goat"]

Hello! I am fairly new to programming. I'm sure it's not a difficult task, but I just can't find the solution.

I need a function in python to store a single random element from list "b" as a new list, but the elements that share the same index with elements from list "a" that are 5 must not be included in the random selection. So in my example, "horse" and "goat" must not be included in the random selection.

Upvotes: 0

Views: 401

Answers (1)

mozway
mozway

Reputation: 262484

You can use a list comprehension and random.choice:

import random

a = [1, 2, 5, 3, 4, 5]
b = ["cat", "dog", "horse", "snake", "elephant", "goat"]
c = [random.choice([e for i,e in zip(a,b) if i != 5])]

Example output: ['dog']

Upvotes: 1

Related Questions