在函数中传递数据帧和变量

Question

我正在尝试传递数据帧和变量函数来绘制一些函数 在python中使用plotnine。我是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
后面的括号内