apache-spark - 在 PySpark 中获取列的名称/别名

Question

我正在定义一个这样的列对象:

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

我知道我可以使用 str(column) 获得完整的表达式.
但是我怎样才能获得列的别名呢？
在这个例子中，我正在寻找一个函数 get_column_name哪里 get_column_name(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'