Question

我创建了此函数，该函数可根据特定年份返回我正在分析的某些国家的人均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

我已经测试了该功能，并且可以正常工作。但是，我想使其更加用户友好，并让用户输入年份（功能的参数）。我该怎么做？

Answer 1

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

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

在Python函数中使用input（）作为参数

1 个答案: