前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >dataframe的select传入不定参数

dataframe的select传入不定参数

作者头像
机器学习和大数据挖掘
发布2019-07-02 09:15:41
1.7K0
发布2019-07-02 09:15:41
举报
文章被收录于专栏:数据挖掘

在提取 dataframe 里面的列时,需要传入不定参数,即 dataframe.select(args) 。例如某个 dataframe 如下:

一般提取某列或者某几列的时候是这样子写的:

代码语言:javascript
复制
dataframe.select("id", "col1", "col2")

但是有需求需要传入不定参数提取不定的列,则可以将需要提取的列放入到一个 Array 中,再如此调用:

代码语言:javascript
复制
dataframe.select(Array.head, Array.tail: _*)

因为 select 官方定义的时候是支持传入不定参数的:

代码语言:javascript
复制
def select(col: String, cols: String*): DataFrame = select((col +: cols).map(Column(_)) : _*)

唯一的要求是 Array 里面元素的类型是 String 类型。

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2018-01-11 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档