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

Question

我已经搜索了一段时间，我确信有人在我之前问过这个问题，但我认为我没有使用正确的搜索词(不知道该怎么称呼这个)。

我有一个包含 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]