Display (Печать) с колоннами Python

Question

У меня есть ниже сценарий

import itertools 
import random 

print "a" 
print "\n" 
def weighted_random(weights): 
    number = random.random() * sum(weights.values()) 
    for k,v in weights.iteritems(): 
     if number < v: 
      break 
     number -= v 
    return k 
weights = {'1': 48, 
      '0': 52} 
    for i in itertools.count(5): 
    print weighted_random(weights) 

print "b" 
print "\n" 
def weighted_random(weights): 
    number = random.random() * sum(weights.values()) 
    for k,v in weights.iteritems(): 
     if number < v: 
      break 
     number -= v 
    return k 
weights = {'1': 53, 
      '0': 50} 
for i in itertools.count(5): 
    print weighted_random(weights) 

Я вижу результат как

a 
1 
0 
1 
1 
0 

b 
1 
0 
0 
1 
1 

Но я хочу сделать это на что-то подобное. Я знаю, что я мог бы назначить a и b во что-то, а затем использовать их, чтобы отображать их в двух столбцах, но пока я не мог этого достичь. Нужна помощь!

a b 
1 0 
0 0 
0 1 
0 1 
1 0 

Answer 1

Во-первых, не определяйте то же самое weighted_random функция дважды.

Затем вы можете вызвать функцию и распечатать ее результат в одном цикле и, следовательно, в двух параллельных столбцах.

import itertools 
import random 

def weighted_random(weights): 
    number = random.random() * sum(weights.values()) 
    for k,v in weights.iteritems(): 
     if number < v: 
      break 
     number -= v 
    return k 


weights_a = {'1': 48, 
      '0': 52} 
weights_b = {'1': 53, 
      '0': 50} 

print "a", "b" 

for i in itertools.count(5): 
    print weighted_random(weights_a), weighted_random(weights_a)