@@ -1584,36 +1584,36 @@ def putmask(self, mask, value):
15841584
15851585 if isinstance (value , (list , tuple , Index , Series )):
15861586 if isinstance (value , (list , tuple )):
1587- pandas_replace_value = pd .Series (value )
1587+ pandas_value = pd .Series (value )
15881588 elif isinstance (value , (Index , Series )):
1589- pandas_replace_value = value .to_pandas ()
1589+ pandas_value = value .to_pandas ()
15901590
1591- if self .size != pandas_replace_value .size :
1591+ if self .size != pandas_value .size :
15921592 # TODO: We can't support different size of value for now.
15931593 raise ValueError ("value and data must be the same size" )
15941594
1595- replace_return_type = as_spark_type (pandas_replace_value .dtype .type )
1595+ replace_return_type = as_spark_type (pandas_value .dtype .type )
15961596
15971597 @pandas_udf (returnType = replace_return_type if replace_return_type else StringType ())
15981598 def replace_pandas_udf (sequence ):
1599- return pandas_replace_value [sequence ]
1599+ return pandas_value [sequence ]
16001600
16011601 sdf = sdf .withColumn (replace_col , replace_pandas_udf (dist_sequence_col_name ))
16021602 else :
16031603 sdf = sdf .withColumn (replace_col , F .lit (value ))
16041604
16051605 if isinstance (mask , (list , tuple , Index , Series )):
16061606 if isinstance (mask , (list , tuple )):
1607- pandas_replace_mask = pd .Series (mask )
1607+ pandas_mask = pd .Series (mask )
16081608 elif isinstance (mask , (Index , Series )):
1609- pandas_replace_mask = mask .to_pandas ()
1609+ pandas_mask = mask .to_pandas ()
16101610
1611- if self .size != pandas_replace_mask .size :
1611+ if self .size != pandas_mask .size :
16121612 raise ValueError ("mask and data must be the same size" )
16131613
16141614 @pandas_udf (returnType = BooleanType ())
16151615 def masking_pandas_udf (sequence ):
1616- return pandas_replace_mask [sequence ]
1616+ return pandas_mask [sequence ]
16171617
16181618 sdf = sdf .withColumn (masking_col , masking_pandas_udf (dist_sequence_col_name ))
16191619 elif not isinstance (mask , list ) and not isinstance (mask , tuple ):
0 commit comments