Я использую Spark 2.2.0 и хотел понять, как функция greatest
работает с отсутствующими данными.
Тем не менее, у меня есть два противоречивых сценария.
import pandas as pd
import pyspark.sql.functions as F
df_pd = pd.DataFrame(
data={
'a': [1, 2, 3],
'b': [-1.0, 0.5, 2.7],
'c': [3, None, 1]})
df1 = spark.createDataFrame(df_pd)
df2 = spark.createDataFrame([(1, -1.0, 3), (2, 0.5, None), (3, 2.7, 1)], ['a', 'b', 'c'])
df1.withColumn('max', F.greatest(F.col('a'), F.col('b'), F.col('c'))).show()
df2.withColumn('max', F.greatest(df2.a, df2.b, df2.c)).show()
Первый дает это:
+---+----+---+---+
| a| b| c|max|
+---+----+---+---+
| 1|-1.0|3.0|3.0|
| 2| 0.5|NaN|NaN|
| 3| 2.7|1.0|3.0|
+---+----+---+---+
Пока второе:
+---+----+----+---+
| a| b| c|max|
+---+----+----+---+
| 1|-1.0| 3|3.0|
| 2| 0.5|null|2.0|
| 3| 2.7| 1|3.0|
+---+----+----+---+
Есть идеи, почему это происходит? В идеале я хочу пропустить нули, и при тестировании я наткнулся на это.