python - 输入提示不会结束

Question

运行 Python 3.6.5

我是 python 的新手。当我在终端中分别运行这些行时，我得到了我想要的。当我运行python文件时，输入提示'ideal weight?'提交号码后不会结束。它不断重复“理想体重？”。我正在尝试从“权重”集中找到将总结为用户输入的数字组合。

import itertools
weights = [3, 3, 7.5, 7.5, 10]
weightint = int(input('ideal weight? '))
result = [seq for i in range(len(weights), 0, -1) for seq in itertools.combinations(weights, i) if sum(seq) == weightint]
print(result)

谁能帮我解释一下我做错了什么。谢谢!

Answer 1

不确定您的终端有什么问题。考虑使用 argparse而不是 input:

import itertools
import argparse

MY_WEIGHTS = [3, 3, 7.5, 7.5, 10]

def find_weight(w):
    result = [seq for i in range(len(MY_WEIGHTS), 0, -1) for seq in itertools.combinations(MY_WEIGHTS, i) if sum(seq) == w]
    return result

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('-w', '--weight', required=True, type=int, help='The weight')
    args = parser.parse_args()
    result = find_weight(args.weight)
    print('result: {}'.format(result))

if __name__ == '__main__':
    main()

然后使用 --weight 或 -w 从命令行调用它:

python3 ./weight.py --weight 10

result: [(10,)]