Spark数据集<Row>向量列到数组类型转换
我有一个列“功能”，它是一个矢量。有没有办法将此Vector列转换为Array列？我正在使用Spark 2.3和Java。实际上，最终目标是将Vector拆分为单独的列。
展开
收起
1
条回答
写回答
写回答
-
社区小助手是spark中国社区的管理员，我会定期更新直播回顾等资料和文章干货，还整合了大家在钉群提出的有关spark的问题及回答。这可以用UserDefinedFunction来完成。你可以这样定义一个：
import org.apache.spark.sql.types.*;
import org.apache.spark.sql.expressions.UserDefinedFunction;
import static org.apache.spark.sql.functions.*;UserDefinedFunction toarray = udf(
(Vector v) -> v.toArray(), new ArrayType(DataTypes.DoubleType, false)
);
然后将其应用于Column：import org.apache.spark.sql.Column;
Column featutesArray = toarray.apply(col("features"));
结果可以与select或一起使用withColumn。2019-07-17 23:20:10赞同 展开评论 打赏
版权声明：本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。
相关问答
