Python 3.x：调用和执行函数的用户输入

Question

有很多类似的问题，但没有一个答案能解决我的问题。

我已经定义了几个解析大型数据集的函数。首先，我调用数据，然后将数据（表示为 .txt 中的行和列）组织到列表中，我将为各个数据条目编制索引。之后，我建立了我的函数，这些函数将一次处理一个列表。代码如下：

f = open(fn)
for line in iter(f):

    entries = [i for i in line.split() if i]

    def function_one():
        if entries[0] == 150:
            # do something
    def function_two():
        if entries[1] == 120:
        # do something else
    def function_three():
        if len(entries) > 10:
        # do something else

等等。等等。

我试图提示用户询问他们想要执行什么函数，因为每个函数返回关于数据集的不同内容。我的尝试如下：

    f_call = input('Enter Function Name: ')
    if f_call in locals().keys() and callable(locals()['f_call']):
        locals()['f_call']()
    else:
        print('Function Does Not Exist')

当我运行脚本时，系统会提示我输入'Enter Function Name:'，如果我输入'function_one' 并返回，则会提示prints'Function Does Not Exist'。我想看看，如果输入正确，脚本将只执行用户输入的函数。如果用户输入正确，该函数应该运行并print 解析的数据。

我也尝试使用dict 来存储函数，但没有成功。

任何帮助将不胜感激。

Answer 1

根据您的 cmets，我认为您正在尝试实现以下目标：

def function_one(data):
    if data[0] == 150:
        pass  # do something

def function_two(data):
    if data[1] == 120:
        pass  # do something else

def function_three(data):
    if len(data) > 10:
        pass # do something entirely different

这定义了接受参数的函数，以便您以后可以重用它们。然后你想问用户在处理数据时使用哪个函数，所以：

while True:  # loop while we don't get a valid input
    user_function = input('Enter a function name to use: ')  # ask the user for input
    if user_function in locals() and callable(locals()[user_function]):  # if it exists...
        user_function = locals()[user_function]  # store a pointer to the function
        break  # break out of the while loop since we have our valid input
    else:
        print('Invalid function name, try again...')

最后，您可以加载您的文件，逐行读取，拆分并按用户决定的功能进行处理：

with open(file_name, "r") as f:
    for line in f:
        entries = line.split()  # no need to check for empty elements
        user_function(entries)  # call the user selected function and pass `entries` to it

当然，你可以在之后做进一步的处理。

更新 - 这是对上述代码的简单测试，给定文件 test_file.txt 包含：

标记化的第 1 行
标记化的第 2 行
标记化的第 3 行

和file_name = "test_file.txt"在文件中定义，而函数定义为：

def function_one(data):
    print("function_one called: {}".format(data))

def function_two(data):
    print("function_two called: {}".format(data))

def function_three(data):
    print("function_three called: {}".format(data))

如果你执行代码，这是输出/跟踪：

输入要使用的函数名称：bad_name_on_purpose
函数名无效，重试...
输入要使用的函数名称：function_two
function_two 调用：['tokenized', 'line', '1']
function_two 调用：['tokenized', 'line', '2']
function_two 调用：['tokenized', 'line', '3']

Answer 2

要使您的代码正常工作，只需在调用时保留变量 f_call 不带 ''

f_call = input('Enter Function Name: ')
if f_call in locals().keys() and callable(locals()[f_call]):
    locals()[f_call]()
else:
    print('Function Does Not Exist')

Answer 3

这可能不是修复它的最有效方法，但您可以使用以下方法：

f_call = raw_input('Enter Function Name: ')
if f_call == "function_one":
    function_one()
if f_call == "function_two":
    function_two()
if f_call == "function_three":
    function_three()
else:
    print('Function Does Not Exist')