Оно работает! Спасибо!
я есть SparkDataFrame в pyspark (2.1.0), и я хочу получить имена только числовых столбцов или только строковых столбцов.
Например, это схема моего DF:
root
|-- Gender: string (nullable = true)
|-- SeniorCitizen: string (nullable = true)
|-- MonthlyCharges: double (nullable = true)
|-- TotalCharges: double (nullable = true)
|-- Churn: string (nullable = true)
Это то, что мне нужно:
num_cols = [MonthlyCharges, TotalCharges]
str_cols = [Gender, SeniorCitizen, Churn]
Как я могу это сделать? Спасибо!