column(colName)
Описание
Функция column()
является синонимом функции col()
и создает объект Column по имени столбца. Это альтернативный способ ссылаться на столбцы в DataFrame.
Параметры
colName
: String - имя столбца
Возвращаемое значение
Column - объект столбца
Пример использования
from pyspark.sql import SparkSession
from pyspark.sql.functions import column
# Создаем SparkSession
spark = SparkSession.builder.appName("column_example").getOrCreate()
# Создаем DataFrame
data = [
("A", 100),
("B", 200),
("C", 300)
]
df = spark.createDataFrame(data, ["name", "value"])
# Используем column() для ссылки на столбцы
result = df.select(
column("name"),
column("value") + 50
).show()
# Результат:
# +----+-----+
# |name|value|
# +----+-----+
# | A| 150|
# | B| 250|
# | C| 350|
# +----+-----+