[jira] [Commented] (SPARK-30989) TABLE.COLUMN reference doesn't work with new columns created by UDF

["hemanth meka (Jira)" <ji...@apache.org>]

    [ https://issues.apache.org/jira/browse/SPARK-30989?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=17057255#comment-17057255 ] 

hemanth meka commented on SPARK-30989:
--------------------------------------

The alias "cat" is defined as a dataframe having 2 columns "x" and "y". The column "z" is generated from "cat" into a new dataframe "df2" but below code works and hence this exception looks like it should be the expected behaviour. is it not?
df2.select("z")
 

> TABLE.COLUMN reference doesn't work with new columns created by UDF
> -------------------------------------------------------------------
>
>                 Key: SPARK-30989
>                 URL: https://issues.apache.org/jira/browse/SPARK-30989
>             Project: Spark
>          Issue Type: Bug
>          Components: SQL
>    Affects Versions: 2.4.4
>            Reporter: Chris Suchanek
>            Priority: Major
>
> When a dataframe is created with an alias (`.as("...")`) its columns can be referred as `TABLE.COLUMN` but it doesn't work for newly created columns with UDF.
> {code:java}
> // code placeholder
> df1 = sc.parallelize(l).toDF("x","y").as("cat")
> val squared = udf((s: Int) => s * s)
> val df2 = df1.withColumn("z", squared(col("y")))
> df2.columns //Array[String] = Array(x, y, z)
> df2.select("cat.x") // works
> df2.select("cat.z") // Doesn't work
> // org.apache.spark.sql.AnalysisException: cannot resolve '`cat.z`' given input 
> // columns: [cat.x, cat.y, z];;
> {code}
> Might be related to: https://issues.apache.org/jira/browse/SPARK-30532



--
This message was sent by Atlassian Jira
(v8.3.4#803005)

---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscribe@spark.apache.org
For additional commands, e-mail: issues-help@spark.apache.org