gpt4 book ai didi

scala - 根据包含列表元素的列值创建 bool 标志

转载 作者:行者123 更新时间:2023-12-04 15:52:32 26 4
gpt4 key购买 nike

<分区>

def myFunction(df: DataFrame): DataFrame = {
val myList= List("a","b","c")

df
.withColumn("myFlag",
if (myList.contains(df.select(col("columnName1")))) lit("true") else lit(false))
}

我想编写一个函数,它接受一个 Dataframe,并向其添加一个名为“myFlag”的列。

如果对应的“columnName1”的值是“myList”的一个元素,我希望“myFlag”为true,否则为false

为简单起见,“columnName1”值和“myList”仅包含字符串。

我上面的功能将不起作用。有什么建议吗?

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com