Передача данных и переменных в функцию

Question

Я пытаюсь передать функции dataframe и variable для построения некоторой функции с использованием plotnine в python.Я новичок в Python.Возможно ли это сделать?Мой код

def countPred_eda(dataset, variableName, targetVariable):
    print(dataset.variableName.describe())
    ggplot(dataset) + \
    aes(targetVariable, variableName, fill = targetVariable) + geom_boxplot(alpha = .8) + \
    labs(x =  targetVariable, y = variableName) + \
    ggtitle("Churn ratio with number_customer_service_calls ")

, называемый

countPred_eda(train_data, number_customer_service_calls,churn)

Answer 1

Предполагается, что dataset имеет тип pd.DataFrame, а variableName и targetVariable - две строки, которые содержат имена двух столбцов в этом кадре данных, вы можете сделать это следующим образом с двумя исключениями:

1. Вместо dataset.variableName.describe() вы должны использовать синтаксис индексации, например dataset[variableName].describe()
2. Параметры ggplot, например, aes, заключены в квадратные скобки непосредственно за dataset