Question

我一直在搜索，我确定有人必须在我面前提出这个问题，但我认为我没有使用正确的搜索词（不知道该怎么称呼 EM>）。

我有一个带有sql文件的目录，我想列出目录并为每个文件分配一个用户随后选择的数字值，并设置一个带有该文件名的变量，以便我可以使用它传递给sql执行功能。我嘲笑下面的代码来展示我脑子里的想法。

import os

upgrades =  os.listdir('upgrades')

select = 1

print '\nPlease select your upgrade: \n'

for sql in upgrades:
    print '['+str(select)+']: '+sql
    select +=1

print '\n'

selection = raw_input('-->')

Answer 1

由于upgrades是一个列表，因此您只需将其编入索引即可。因为您从1开始，列表索引以0开头，所以只需执行：

selection = int(raw_input('-->'))
selected_sql = upgrades[selection - 1]

使用列表中的值以交互方式设置变量

1 个答案: