如何从 F.col 对象恢复列名？

Question

简单的问题：假设我们

import pyspark.sql.functions as F

那么如何从 pyspark.sql.column.Column object 恢复列名字符串 'a' F.col('a')。

例如，如果我们输入str(F.col('a'))，我们有

而不是原始的列名称“a”。

Answer 1

没有受支持的方法可以执行此操作。你可以稍微修改一下并尝试围绕这些线进行一些操作

from pyspark.sql.functions import lit, col
from pyspark.sql.column import Column
from py4j.protocol import Py4JError

def name(self):
    try:
        return self._jc.expr().name()
    except Py4JError:
        return None

Column.name = property(name)

这将给出：

from pyspark.sql.functions import lit, col

col("a").name 

'a'

lit(1).name is None

True