熟悉pandas的pythoner 应该知道给dataframe增加一列很容易，直接以字典形式指定就好了，pyspark中就不同了，摸索了一下，可以使用如下方式增加
代码语言：javascript复制from pyspark import SparkContext
from pyspark import SparkConf
from pypsark.sql import SparkSession
from pyspark.sql import functions
spark = SparkSession.builder.config(conf=SparkConf()).getOrCreate()
data = [['Alice', 19, 'blue', '["Alice", 19, "blue"]'],
['Jane', 20, 'green', '["Jane", 20, "green"]'],
['Mary', 21, 'blue', '["Mary", 21, "blue"]'], ]
frame = spark.createDataFrame(data, schema=["name", "age", "eye_color", "detail"])
frame.cache()
frame.show()—– — ——— ——————– | name|age|eye_color| detail| —– — ——— ——————– |Alice| 19| blue|[“Alice”, 19, “bl…| | Jane| 20| green|[“Jane”, 20, “gre…| | Mary| 21| blue|[“Mary”, 21, “blue”]| —– — ——— ——————–
1、 增加常数项
代码语言：javascript复制frame2 = frame.withColumn("contant", functions.lit(10))
frame2.show()—– — ——— ——————– ——- | name|age|eye_color| detail|contant| —– — ——— ——————– ——- |Alice| 19| blue|[“Alice”, 19, “bl…| 10| | Jane| 20| green|[“Jane”, 20, “gre…| 10| | Mary| 21| blue|[“Mary”, 21, “blue”]| 10| —– — ——— ——————– ——-
2、简单根据某列进行计算
2.1 使用 withColumn
代码语言：javascript复制frame3_1 = frame.withColumn("name_length", functions.length(frame.name))
frame3_1.show()—– — ——— ——————– ———– | name|age|eye_color| detail|name_length| —– — ——— ——————– ———– |Alice| 19| blue|[“Alice”, 19, “bl…| 5| | Jane| 20| green|[“Jane”, 20, “gre…| 4| | Mary| 21| blue|[“Mary”, 21, “blue”]| 4| —– — ——— ——————– ———–
2.2 使用 select
代码语言：javascript复制frame3_2 = frame.select(["name", functions.length(frame.name).alias("name_length")])
frame3_2.show()—– ———– | name|name_length| —– ———– |Alice| 5| | Jane| 4| | Mary| 4| —– ———–
2.3 使用 selectExpr
代码语言：javascript复制frame3_3 = frame.selectExpr(["name", "length(name) as name_length"])
frame3_3.show()—– ———– | name|name_length| —– ———– |Alice| 5| | Jane| 4| | Mary| 4| —– ———–
3、定制化根据某列进行计算
比如我想对某列做指定操作，但是对应的函数没得咋办，造，自己造~
代码语言：javascript复制frame4 = frame.withColumn("detail_length", functions.UserDefinedFunction(lambda obj: len(json.loads(obj)))(frame.detail))
# or
def length_detail(obj):
return len(json.loads(obj))
frame4 = frame.withColumn("detail_length", functions.UserDefinedFunction(length_detail)(frame.detail))
frame4.show()—– — ——— ——————– ————- | name|age|eye_color| detail|detail_length| —– — ——— ——————– ————- |Alice| 19| blue|[“Alice”, 19, “bl…| 3| | Jane| 20| green|[“Jane”, 20, “gre…| 3| | Mary| 21| blue|[“Mary”, 21, “blue”]| 3| —– — ——— ——————– ————-
到此这篇关于pyspark给dataframe增加新的一列的实现示例的文章就介绍到这了,更多相关pyspark dataframe增加列内容请搜索ZaLou.Cn以前的文章或继续浏览下面的相关文章希望大家以后多多支持ZaLou.Cn！
