如何从scala中的org.apache.spark.sql.Column类型获取键并放入列表变量中？

Question

我正在尝试从 org.apache.spark.sql.Column 类型变量获取键并将其放入列表中，以便我可以进行一些模式比较。

inputFieldMap: org.apache.spark.sql.Column = keys:[customerID,country],values[customerID,country]

expected output
newvariable: list[string]=[customerID,country];

Answer 1

使用

map_keys

函数从

map

数据类型的列中获取键。

df.selectExpr("map_keys(inputFieldMap) as newvariable")