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

Question

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

谢谢

Answer 1

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

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)