python - 如何使用'def'获取输入值

Question

如果我使用全局变量。我可以这样写。

while True:
    name_str = input('Please give me the file name:')
    if name_str == 'table.csv':
        break
    else:
        print('Bad file name, please try again.')

如何定义一个函数并从此函数获取输入？在开始函数中获取用户输入，调用main函数获取用户输入。

def get_name(n):
    x = input('Please give me the file name:')
    ...
    ...

def main():
    ...
    ...

Answer 1

你想把它放在一个文件（test.py）中：

def get_input():
    return raw_input('get file: ')

if __name__ == '__main__':
    print get_input()

然后调用它：

$蟒蛇测试.py

注意两点：

1. input() vs raw_input() -- input()不安全
2. 在脚本环境中使用成语“if name ...”而不是具有 main() 函数。

Answer 2

有点取决于您要完成的工作。

如果你的方法是从用户那里收集信息，你可以这样写：

def get_name():
    return input('Please give me the file name. ')

如果您的方法正在解析来自用户的信息，它将接受文件名作为参数并编写如下内容：

def check_file_name(fname):
    if fname == 'table.csv':
        return True
    return False