擅长:python、mysql、java
<p>虽然使用pandas数据帧有一个很好的答案,但是既然你在标签中提到了pyspark,我想你是指spark dataframes?在</p>
<p>如果是这样,您可以这样做:</p>
<pre><code>import pyspark.sql.functions.when,lit
newDF = df.withColumn("col10", when(df["col9"] < 0.5, lit("A")).otherwise(when(df["col9"] > 0.75, lit("C")).otherwise(lit("B"))
</code></pre>
<p>我假设列的合法值是0-1,但是如果需要显式地检查它们,只需更改条件并为非法值添加额外的when/other</p>