请教一个程序的写法～谢谢～

想写个程序，感觉好像不难，但是想了很久都没什么思路，希望大侠们指点一下，谢谢！～
创建一个矩阵a=[[a11,a12,a13],[a21,a22,a23],[a31,a32,a33]]，另外再创建一个大小相同的矩阵b=[[0,1,0],[0,0,1],[1,0,0]]，其中只有0和1两个元素而且是随机生成的。然后当b中的元素为1的时候打印a中的相对应的元素；当b中元素为0的时候不输出任何值。
呵呵，因为刚开始学编程，很多时候都没什么思路～

for i in range(len(b)):
     for j in range(len(b[i])):
             if b[i][j] == 1:
                     print a[i][j]

tag = []
for I in b:
        for i in I:
                if i = '1':
                        tag.append((b.index(I),l.index(i)))
for (x,y) in tag:
        print a[x][y]

谢谢两位，但是里面b中的元素0和1是随机生成的，这个该怎么写啊？

import random
b=random.randint(0,1)