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

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)