【发布时间】:2015-08-05 22:07:25
【问题描述】:
我在 Spark SQL 中有 join 函数。此函数需要一个连接条件,如果我们要连接的列名称不同,则需要将它们作为连接表达式传递。
例子:
x.join(y, x.column1 == y.column2)
这意味着我们在 x 和 x 中的 column1 和 column2 中加入数据帧 x 和 y y
我想编写一个函数,它将两个数据框的列名作为参数并连接这些列。问题是连接表达式不能是字符串。我看过像one 这样的问题,其中使用映射来映射变量名,但这不符合我的需要。我需要删除使列名成为字符串的引号并将它们传递给join 函数。
我已经检查过,如果我们加入的列没有相同的名称(除了生成具有新列名的数据框之一的副本之外),在 Pyspark 中没有其他方法可以执行此操作。这是因为数据框是不可变的,不能更改列名)
还有其他方法可以将列名传递到连接表达式中吗?
【问题讨论】:
-
你能试试
x.join(y, getattr(x, 'column1') == getattr(y, 'column2'))吗? -
@hgwells 当我这样做时,我得到了实际的列:
type(getattr(x, 'column1')) <class 'pyspark.sql.column.Column'> -
x.column1的类型是什么?我的猜测是,这也是<class 'pyspark.sql.column.Column'>类型 -
@hgwells 它也是一个列类型:\抱歉怀疑
-
你让它工作了吗?
标签: python join argument-passing apache-spark-sql pyspark