pyspark dataframe 自定義分區器

 def myp(x):
 	return x % 100

 pp = F.udf(myp)
 df = spark.range(900)
 df.show()
 df = df.repartitionByRange(pp('id'))
 df.write.format('csv').save('output')

把 id%100 相等的數據放到一個partition