Nov-01-2019, 10:34 PM
Hi, could someone help me on this code plz ? It says"IndexError: List index out of range issue" in "n = int(sys.argv[1])". btw i m using python 3.7 if it can help.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
import sys import numpy as np import matplotlib import matplotlib.pyplot as plt from matplotlib.ticker import FuncFormatter def multiplication_table(n, N = None , number_labels = True ): """Create and plot an image of a multiplication table modulo n The table is of ij % n for i, j = 1, 2, ..., N-1. If not supplied, N defaults to n. If N is a mutiple of n, the pattern is repeated across the created image. The "rainbow" colormap is used, but zeros (corresponding to factors of n) are displayed in white. """ if not N: N = n # A multiplication table (modulo n) arr = np.fromfunction( lambda i,j:(i + 1 ) * (j + 1 ) % n, (N - 1 ,N - 1 )) # Select a colormap, but we'll set 0 values to white cmap = matplotlib.cm.get_cmap( 'rainbow' ) cmap.set_under( 'w' ) fig, ax = plt.subplots() # Plot an image of the multiplication table in colours for values greater # than 1. Zero values get plotted in white thanks to set_under, above. ax.imshow(arr, interpolation = 'nearest' , cmap = cmap, vmin = 1 ) # Make sure the tick marks are correct (start at 1) tick_formatter = FuncFormatter( lambda v, pos: str ( int (v + 1 ))) ax.xaxis.set_major_formatter(tick_formatter) ax.yaxis.set_major_formatter(tick_formatter) # For small n, write the value in each box of the array image. if number_labels and N < 21 : for i in range (N - 1 ): for j in range (N - 1 ): ax.annotate(s = str ((i + 1 ) * (j + 1 ) % n), xy = (i,j), ha = 'center' , va = 'center' ) # The user supplies n (and optionally N) as command line arguments n = int (sys.argv[ 1 ]) try : N = int (sys.argv[ 2 ]) except IndexError: N = None multiplication_table(n, N, False ) plt.show() |