Renvoi d'un élément aléatoire à partir d'un tableau Python basé sur des critères de recherche

Question

Excuses si cela est simple, mais j'ai cherché un peu de temps maintenant et ne peux pas trouver une solution simple et efficace.

J'ai une liste Python bidimensionnelle de listes qui ne comprend que des 1 et des 0.

par exemple:

a=[[0,1,0],[0,1,1],[1,0,1]] 

Je voudrais revenir, au hasard, les indices d'un élément aléatoire qui est = 1. Dans ce cas, je voudrais revenir soit:

[0,1], [1,1], [1,2], [2,0], or [2,2] 

avec une probabilité égale.

Je pourrais parcourir tous les éléments de la structure et compiler une liste d'indices admissibles, puis en choisir un au hasard en utilisant random.choice (liste) - mais cela semble très lent et je ne peux pas m'empêcher de penser qu'il y a un plus , plus de façon Pythonienne d'aborder cela. Je le ferai probablement pour un tableau 20x20 et je devrai le faire plusieurs fois, donc je pourrais le faire aussi efficacement que possible.

Merci d'avance pour toute aide et conseil!

Answer 1

j'utiliser une compréhension de la liste pour générer une liste de tuples (positions de 1), puis random.choice:

from random import choice 

a = [[0,1,0],[0,1,1],[1,0,1]] 
mylist = [] 

[[mylist.append((i,j)) for j, x in enumerate(v) if x == 1] for i, v in enumerate(a)] 
print(choice(mylist)) 

Answer 2

Lorsque vous obtenez votre résultat de vérifier random.choice si elle est comment vous voulez avec les éléments corrects si elle est pas aléatoire à nouveau

def return_random(li): 
    item = random.choice(li) 
    if item == 1: #insert check here 
     return item 
    else: 
     return_random(li) 

Edit: pour éviter toute confusion avec le module re, grâce

Answer 3

Je voudrais utiliser un tableau NumPy à y parvenir:

from numpy import array 
random_index = tuple(random.choice(array(array(a).nonzero()).T)) 

Si votre magasin vos données dans des tableaux numpy dès le début, cette approche sera probablement plus rapide que tout ce que vous pouvez faire avec une liste de listes.

Si vous voulez choisir plusieurs indices pour les mêmes données, il y a des approches encore plus rapides.

Answer 4

random.choice nous permettent de choisir un élément au hasard dans une liste, donc nous avons juste besoin d'utiliser une compréhension de liste pour créer une liste des indices où les éléments sont 1 et puis choisir un au hasard.

Nous pouvons utiliser la compréhension de la liste de suivi:

>>> a = [[0,1,0],[0,1,1],[1,0,1]] 
>>> [(x,y) for x in range(len(a)) for y in range(len(a[x])) if a[x][y] == 1] 
[(0, 1), (1, 1), (1, 2), (2, 0), (2, 2)] 

Ce qui signifie que nous pouvons faire:

>>> import random 
>>> random.choice([(x,y) for x in range(len(a)) for y in range(len(a[x])) if a[x][y] == 1]) 
(1, 1) 

Si vous allez faire cela plusieurs fois il peut être utile mise en cache la liste des index générés par la compréhension, puis en choisissant plusieurs fois plutôt que de calculer la compréhension de la liste à chaque fois.

Answer 5

Une autre idée serait de stocker les données d'une manière complètement différente: Au lieu d'une liste de listes, utilisez un ensemble de paires d'index représentant les entrées qui sont 1.Dans votre exemple, ce serait

s = set((0, 1), (1, 1), (1, 2), (2, 0), (2, 2)) 

Pour choisir au hasard une paire d'index, utilisez

random.choice(list(s)) 

Pour définir une entrée à 1, utilisez

s.add((i, j)) 

Pour définir une entrée à 0 , utilisez

s.remove((i, j)) 

Pour retourner une entrée, utilisez

s.symmetric_difference_update([(i, j)]) 

Pour vérifier si une entrée est 1, utilisez

(i, j) in s