PySpark:如何删除DataFrame中的非数字列？

from pyspark.sql.functions import col, isnan

def drop_non_numeric_columns(df):
    numeric_columns = [column for column in df.columns if df.select(column).dtypes[0][1] in ['double', 'float', 'integer']]
    df = df.select(*numeric_columns)
    return df

# 使用示例
df = spark.createDataFrame([(1, 'a', 2.0), (3, 'b', 4.0)], ['col1', 'col2', 'col3'])
df = drop_non_numeric_columns(df)
df.show()