get_json_obj _fails для SelectExpr (), но работает для Select в Pyspark

Question

У меня странная проблема, я пытаюсь отобразить значения моего объекта JSON, он отлично работает с select(), но не работает с selectExp(), я получаю странную ошибку, следующую в моей реализации,

from pyspark.sql import SparkSession
from pyspark.sql.functions import *
from pyspark.sql.functions import col

spark = SparkSession.builder.appName("JsonPractice").getOrCreate()
my_json_df = spark.range(1).selectExpr(
    """'{"sample_json":{"sample_json1":["1st_vale","2nd_val"]}}' as my_json_column""")
my_json_df.selectExpr(get_json_object(col("my_json_column"), "$.sample_json.sample_json1[1]")).show(2)
my_select_expr = get_json_object(col('my_json_column'), '$.sample_json.sample_json1')
my_json_df.selectExpr(my_select_expr).show()

Я получаю следующую ошибку

поднять TypeError («Столбец не повторяется») TypeError: столбец не повторяется

Answer 1

Нам не нужно указывать col при использовании selectExpr

my_select_expr = "get_json_object(my_json_column, '$.sample_json.sample_json1')"
my_json_df.selectExpr(my_select_expr).show(10,False)

#or    
my_json_df.selectExpr("get_json_object(my_json_column,'$.sample_json.sample_json1')").show(10,False)
#+-----------------------------------------------------------+
#|get_json_object(my_json_column, $.sample_json.sample_json1)|
#+-----------------------------------------------------------+
#|["1st_vale","2nd_val"]                                     |
#+-----------------------------------------------------------+

UPDATE:

from pyspark.sql.functions import *
my_select_expr=get_json_object(col('my_json_column'),'$.sample_json.sample_json1')
my_json_df.select(my_select_expr).show(10,False)
#+-----------------------------------------------------------+
#|get_json_object(my_json_column, $.sample_json.sample_json1)|
#+-----------------------------------------------------------+
#|["1st_vale","2nd_val"]                                     |
#+-----------------------------------------------------------+