PySpark字符串列可匹配整数过滤?与Pandas行为差异及方案
PySpark字符串列与整数过滤的差异问题解答
1. 为何PySpark的过滤函数能用整数匹配字符串?
PySpark底层依赖的SQL执行引擎会自动做隐式类型转换:当你用整数和字符串类型的数字列比较时,Spark会尝试将字符串列的内容转换为对应的数值类型(只要字符串是合法的数字格式),再执行相等性判断。比如例子里的字符串"1"会被转成整数1,因此能匹配过滤条件中的整数1。
而Pandas采用严格的类型匹配逻辑,字符串类型的"1"和整数类型的1属于不同类型,直接比较会返回False,所以过滤后得到空DataFrame。
2. 能否在PySpark中进行类型严格过滤,得到与Pandas一致的结果?
可以,通过Spark的**严格相等操作符===**就能实现,它不会触发隐式类型转换,当两边类型不一致时直接返回False:
df.filter(col('number') === 1)
执行这个过滤后,会得到空的DataFrame,和Pandas中pandas_df[pandas_df['number']==1]的结果完全一致。
如果你想明确保证过滤条件与列类型一致,也可以将整数条件转为字符串,但这种方式会匹配到字符串"1"的行,和Pandas的结果不同——只有使用===才能得到和Pandas相同的空结果。
内容的提问来源于stack exchange,提问作者TiTo
相关产品推荐
相关产品推荐

