Получить имя / псевдоним столбца в PySpark

Question

Я определяю объект столбца следующим образом:

column = F.col('foo').alias('bar')

Я знаю, что могу получить полное выражение, просто выполнив str(column).Но есть ли способ получить значение только для псевдонима столбца?

В этом примере я ищу функцию FN, где FN(column) возвращает bar.

Answer 1

Один из способов - регулярные выражения:

from pyspark.sql.functions import col
column = col('foo').alias('bar')
print(column)
#Column<foo AS `bar`>

import re
print(re.findall("(?<=AS `)\w+(?=`>$)", str(column)))[0]
#'bar'