UDF返回数据的例子
spark = SparkSession.builder. \ appName("wk_stg_10650018_cs"). \ enableHiveSupport(). \ config("spark.debug.maxToStringFields", "100"). \ getOrCreate() df = spark.createDataFrame([('a', [1, 2, 3],['x','y']), ('b', [5, 2, 3],['x','y'])], ['id', 'c1','c2']) df.createOrReplaceTempView('t1') def gen(begin,end): return list(range(begin,end))
# spark.udf.register('gen', gen, ArrayType(IntegerType())) spark.sql('select id, gen(1,3) as arr from t1').createOrReplaceTempView('t1') spark.sql( "select id,c" " from t1" " lateral view explode(arr) A as c" ).show()
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步