我想检查pyspark dataframe中数组的最后两个值是否为 [1, 0]
并将其更新为 [1, 1]
输入Dataframe
Column1 Array_column
abc [0,1,1,0]
def [1,1,0,0]
adf [0,0,1,0]
输出Dataframe
Column1 Array_column
abc [0,1,1,1]
def [1,1,0,0]
adf [0,0,1,1]
3条答案
按热度按时间bfrts1fy1#
你可以切片数组,做一个
case when
对于最后两个元素,并使用concat
.11dmarpk2#
可以将数组函数与when表达式组合使用:
aemubtdh3#