Используйте input () в качестве аргумента в функции в Python

Question

Я создал эту функцию, которая возвращает ВВП на душу населения и военные расходы на душу населения для некоторых стран, которые я анализирую, исходя из конкретного года.

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)