Adding condition in list in python-CodePudding

I don't know how to add extra conditions to this code. (Only using randint)

For example with list [1, 2, 3, 4, 0] I need to generate random number except the last one (covered in my code) but the next condition is, it can not choose an index number which has a value of 0. So for list [3, 3, 0, 3, 3, 7] it can only consider indexes 0,1,3,4 (not 2 and 5 because I can not include the last number and index with value 0).

My code so far:

import random
    
our = [2, 3, 4, 0]
 
random_index = random.randint(0, len(our)-2)

random_number = our[random_index]

print(random_number)

I will be very glad for any help.

CodePudding user response：

You can create a second list that stores the valid index values.

import random

our = [3, 3, 0, 3, 3, 7]

index = []

for i in range(0, len(our)-1) :
    if our[i] != 0 :
        index.append(i)

# index: [0, 1, 3, 4]

random_index = random.choice(index)

EDIT: You can perform a sanity check for a non-zero value being present.

The any() function returns True if any element of an iterable is True. 0 is treated as False and all non-zero numbers are True.

valid_index = any(our[:-1])

if valid_index:
    index = []
    for i in range(0, len(our)-1) :
        if our[i] != 0 :
            index.append(i)

CodePudding user response：

You can use a while loop to check if the number equals 0 or not.

import random


our = [3, 6, 2, 0, 3, 0, 5]

random_number = 0
while random_number == 0:
    random_index = random.randint(0, len(our)-2)
    random_number = our[random_index]

print(random_number)