Как выбрать столбец в кадре данных по номеру вместо его имени

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)