@@ -1590,36 +1590,36 @@ def putmask(self, mask, value):
15901590
15911591 if isinstance (value , (list , tuple , Index , Series )):
15921592 if isinstance (value , (list , tuple )):
1593- pandas_replace_value = pd .Series (value )
1593+ pandas_value = pd .Series (value )
15941594 elif isinstance (value , (Index , Series )):
1595- pandas_replace_value = value .to_pandas ()
1595+ pandas_value = value .to_pandas ()
15961596
1597- if self .size != pandas_replace_value .size :
1597+ if self .size != pandas_value .size :
15981598 # TODO: We can't support different size of value for now.
15991599 raise ValueError ("value and data must be the same size" )
16001600
1601- replace_return_type = as_spark_type (pandas_replace_value .dtype .type )
1601+ replace_return_type = as_spark_type (pandas_value .dtype .type )
16021602
16031603 @pandas_udf (returnType = replace_return_type if replace_return_type else StringType ())
16041604 def replace_pandas_udf (sequence ):
1605- return pandas_replace_value [sequence ]
1605+ return pandas_value [sequence ]
16061606
16071607 sdf = sdf .withColumn (replace_col , replace_pandas_udf (dist_sequence_col_name ))
16081608 else :
16091609 sdf = sdf .withColumn (replace_col , F .lit (value ))
16101610
16111611 if isinstance (mask , (list , tuple , Index , Series )):
16121612 if isinstance (mask , (list , tuple )):
1613- pandas_replace_mask = pd .Series (mask )
1613+ pandas_mask = pd .Series (mask )
16141614 elif isinstance (mask , (Index , Series )):
1615- pandas_replace_mask = mask .to_pandas ()
1615+ pandas_mask = mask .to_pandas ()
16161616
1617- if self .size != pandas_replace_mask .size :
1617+ if self .size != pandas_mask .size :
16181618 raise ValueError ("mask and data must be the same size" )
16191619
16201620 @pandas_udf (returnType = BooleanType ())
16211621 def masking_pandas_udf (sequence ):
1622- return pandas_replace_mask [sequence ]
1622+ return pandas_mask [sequence ]
16231623
16241624 sdf = sdf .withColumn (masking_col , masking_pandas_udf (dist_sequence_col_name ))
16251625 elif not isinstance (mask , list ) and not isinstance (mask , tuple ):
0 commit comments