时间戳

首页 > 解决方案 > 在 Python 函数中使用 input() 作为参数

python - 在 Python 函数中使用 input() 作为参数

问题描述

我创建了这个函数，它返回我基于特定年份分析的一些国家的人均 GDP 和人均军费开支。

def compare(year):
    simple = data.loc[year, :].sort_values(by = "gdp_pc", ascending = False)
    simple = simple.reset_index()
    simple = simple.drop(["gdp_share", "military_exp", "pop", "gdp"], axis = 1)
    simple = simple.round(2)
    simple.columns = ["Countries", "GDP per capita", "Military spending per capita"]
    simple.index = simple.index + 1

    return simple

我已经测试了这个功能，它工作正常。但是，我想让它更加用户友好并让用户输入年份（函数的参数）。我该怎么做呢？

标签： pythonfunctioninput

解决方案

定义一个包装函数；让我们compare做它的事情（实际比较），让另一个函数负责调用input并将响应传递给compare.

def do_comparison():
    year = int(input("Enter a year to compare against: "))
    return compare(year)

推荐阅读