【问题标题】:How to pass an argument to a function that doesn't take string (Pyspark)如何将参数传递给不带字符串的函数(Pyspark)
【发布时间】:2015-08-05 22:07:25
【问题描述】:

我在 Spark SQL 中有 join 函数。此函数需要一个连接条件,如果我们要连接的列名称不同,则需要将它们作为连接表达式传递。

例子:

x.join(y, x.column1 == y.column2)

这意味着我们在 xx 中的 column1column2 中加入数据帧 xy 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


【解决方案1】:

重新发布我的评论作为将来参考的答案。您可以使用gettatr 函数获取类或模块的任何属性。

x.join(y, getattr(x, 'column1') == getattr(y, 'column2'))

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2011-05-12
    • 2021-10-28
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2011-06-29
    • 2020-01-01
    • 1970-01-01
    相关资源
    最近更新 更多