zufällig 2d Matrix mit Bedingungen füllen

Question

Hier ist meine Situation, und ich bin nur werfen es da draußen, um Ideen zu bekommen, wie man es angehen.

Ich habe Kategorien categories = [0, 1, 3] und eine 10 x 5 Matrix, A

Wie kann ich Matrix A mit zufälliger Auswahl von Kategorien, wie füllen, dass jede Spalte ein und nur ein Vorkommen von ‚0‘ und einem und nur ein Vorkommen haben muss von '1'?

Mein Denkprozess ist:

For each column, Cj, 
    create list L = selection of random integers between [0,9] (no replacement) 
    Set Cj[k[0]] = 0 and Cj[k[1]] = 1 
    Set Cj[k[i]] , i=2, ..,9 

Hat jemand einen besseren Weg, dieses Problem zu lösen?

Answer 1

Hier ist ein Weg, es zu tun vorausgesetzt, Sie haben bereits die Matrix. (Die Frage sagt eine Matrix füllen, schaffen keine Matrix)

import pprint 
import random 

categories = [0,1,3] #complete pool of numbers 
restrictedNumbers = [0, 1] #numbers to only appear once 
unrestrictedNumbers = [num for num in categories if num not in restrictedNumbers] #can appear any number of times 

matrix = [[0, 0, 0, 0, 0] for l in range(10)] #10 rows, 5 columns matrix of zeros 
newMatrix = [] 

for row in matrix: 
    tempRow = restrictedNumbers #first add the restricted numbers since they must appear once 
    while len(tempRow) < len(row): #keep adding numbers from unrestricted till lengths are equal 
     tempRow.append(random.choice(unrestrictedNumbers)) 
    random.shuffle(tempRow) 
    newMatrix.append(tempRow[:]) #finally add the shuffled row to newMatrix 
pprint.pprint(newMatrix) 

Diese den Ausgang gibt: (natürlich wird wegen zufälliger variieren)

[[3, 1, 3, 0, 3], 
[3, 3, 0, 1, 3], 
[0, 3, 1, 3, 3], 
[0, 3, 1, 3, 3], 
[3, 3, 0, 3, 1], 
[0, 1, 3, 3, 3], 
[1, 0, 3, 3, 3], 
[1, 3, 0, 3, 3], 
[3, 3, 0, 3, 1], 
[3, 0, 1, 3, 3]] 

Sie können jede Zeile sehen eine einzelne hat 0 und Single ein

Hier ist ein noch kürzerer Weg:

import pprint 
import random 

categories = [0,1,3] #complete pool of numbers 
restrictedNumbers = [0, 1] #numbers to only appear once 
unrestrictedNumbers = [num for num in categories if num not in restrictedNumbers] #can appear any number of times 

matrix = [[0, 0, 0, 0, 0] for l in range(10)] #10 rows, 5 columns matrix of zeros 
newMatrix = [] 
for row in matrix: 
    row = restrictedNumbers + [random.choice(unrestrictedNumbers) for _ in range(len(row)-len(restrictedNumbers))] 
    random.shuffle(row) 
    newMatrix.append(row) 
pprint.pprint(newMatrix) 

Und sogar kürzer als die (diese eine Zeile wäre, wenn random.shuffle die Liste zurückgegeben statt anstelle von schlurfenden)

import pprint 
import random 

categories = [0,1,3] #complete pool of numbers 
restrictedNumbers = [0, 1] #numbers to only appear once 
unrestrictedNumbers = [num for num in categories if num not in restrictedNumbers] #can appear any number of times 

matrix = [[0, 0, 0, 0, 0] for l in range(10)] #10 rows, 5 columns matrix of zeros 
def returnShuffled(startList): 
    random.shuffle(startList) 
    return startList 

newMatrix = [returnShuffled(restrictedNumbers + [random.choice(unrestrictedNumbers) for _ in range(len(row) - len(restrictedNumbers))]) for row in matrix] 
pprint.pprint(newMatrix) 

Answer 2

Ihre Frage ist ein bisschen vage. Bitte geben Sie weitere Informationen darüber, was Sie in anderen Positionen haben können. wenn eine Position 0 ist und 1 Position 1 ist und die anderen Positionen beliebig sein können, dann benutze diese.

machen Sie ein Array [number_of_columns] = {0,1,2,2, ...., 2], Mischen Sie es. dann für jede Position in der Spalte (i) if (array[i] = 0){array_you_are_making[i]=0};if(array[i] = 1){array_you_are_making[i]=1};if(array[i] = 2){array_you_are_making[i]="do something"};

Klingt wie ein Fall für eine Python shuffle() -Methode.