کد جدول ضرب به پایتون

با سلام

جواب تمرین کدجدول ضرب به زبان پایتون به صورت زیر می شه.

def jadval_zarb(rows, columns):
	for i in range(1,rows+1):
		for j in range(1,columns+1):
			#print(i*j, end=' ')
			print('{:>4}'.format(i* j), end=' ')
		print()

def do_continue():
	choice = str(input('continue(y/n)?')).lower()
	if(choice !='y'):
		return False
	return True

####################
while True:
	rows = int(input("Enter rows:"))
	columns = int(input("Enter columns:"))
	jadval_zarb(rows, columns)
	if(not do_continue()):
		break

یک تابع تولید جدول ضرب هست

و تابع دیگر برای اینکه آیا کاربر مایل به ادامه برنامه می باشد یا نه.