scala – 如何访问数组列中的值？

发布时间：2020-12-16 18:53:22 所属栏目：安全来源：网络整理

导读：我有一个包含一列的Dataframe.该列的每一行都有一个String值数组：我的Spark 2.2 Dataframe中的值 ["123","abc","2017","ABC"]["456","def","2001","ABC"]["789","ghi","DEF"]org.apache.spark.sql.DataFrame = [col: array]root|-- col: array (nullable =

我有一个包含一列的Dataframe.该列的每一行都有一个String值数组：

我的Spark 2.2 Dataframe中的值

["123","abc","2017","ABC"]
["456","def","2001","ABC"]
["789","ghi","DEF"]

org.apache.spark.sql.DataFrame = [col: array]

root
|-- col: array (nullable = true)
|    |-- element: string (containsNull = true)

访问数组中元素的最佳方法是什么？例如,我想在2017年的第四个元素中提取不同的值(回答“ABC”,“DEF”).

解决方法

df.where($"col".getItem(2) === lit("2017")).select($"col".getItem(3))

从https://spark.apache.org/docs/latest/api/scala/index.html#org.apache.spark.sql.Column看到getItem

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!