gen_rand_matrix.py

#! /usr/bin/python                                                                                                                         

import random
import sys

def main():

    ''' generates a random sys.argv[1] x sys.argv[2] matrix.                                                                               
        useful for testing solutions to                                                                                                    
        http://github.com/AlgorithmsNYC/AlgorithmsNYC/tree/master/Problem_0001/ '''

    if len(sys.argv) is not 3:
        print "Usage: gen_rand_matrix.py rows columns"
        sys.exit(1)
    for i in range(0, int(sys.argv[1])):
        for j in range(0, int(sys.argv[2])):
            print random.randint(0, 9),
        print

if __name__ == '__main__' : main()