如何通过编号而不是名称在数据框中选择列

我想通过编号而不是名称在Spark数据框中选择一个列。有可能吗?

谢谢

hz000005 回答:如何通过编号而不是名称在数据框中选择列

如果您要为此编写自己的方法,可以执行以下操作:

package utils

object Extensions {
    implicit class DataFrameExtensions(df: DataFrame) {
        def selecti(indices: Int*) = {
            val cols = df.columns
            df.select(indices.map(cols(_)):_*)
        }
    }
}

现在您可以按以下方式导入并使用此方法:

import utils.Extensions._

df.selecti(1,2,3)
本文链接:https://www.f2er.com/3165436.html

大家都在问