Вам необходимо использовать индекс choose
, чтобы выбрать элемент из клавиши record
с помощью
print(record[int(choose)-1])
Таким образом, код изменится на
list1 = ['ID', 'Customers', 'Employees', 'Orders']
def display(list1):
counter = 0
record = []
for tables in list1:
counter += 1
record.append(tables)
print("%s. %s" % (counter, tables))
return record
def get_list(record):
print("\nPick a number:")
choose = input()
choice = int(choose)-1
print(record[choice])
record = display(list1)
get_list(record)
Вы также можете упростить свой код следующим образом
li = ['ID', 'Customers', 'Employees', 'Orders']
def display(li):
#Iterate through the list using enumerate and print
for idx, tables in enumerate(li):
print("%s. %s" % (idx+1, tables))
def get_list(li):
choose = int(input("\nPick a number:"))-1
#If choose is not a valid index in list, print error and return empty string
if choose < 0 or choose > (len(li)-1):
print('Invalid Choice')
return ''
#Else return chosen string
return li[choose]
display(li)
print(get_list(li))
Выход будет тогда
1. ID
2. Customers
3. Employees
4. Orders
Pick a number:
3
Employees
1. ID
2. Customers
3. Employees
4. Orders
Pick a number:0
Invalid Choice