使用Pyspark将数据转换为DataFrame。旧的提取使用了以下SQL行：

case when location_type = 'SUPPLIER' then SUBSTRING(location_id,1,length(location_id)-3)

我引入数据并将其加载到DF中，然后尝试使用以下代码进行转换：

df = df.withColumn("location_id", F.when(df.location_type == "SUPPLIER",
                         F.substring(df.location_id, 1, length(df.location_id) - 3))
                         .otherwise(df.location_id))`

substring方法接受int作为第三个参数，但是length()方法给出了一个Column对象。我没有运气尝试强制转换它，也没有找到接受Column的方法。还尝试使用expr() Package 器，但是同样无法使其工作。
供应商ID看起来像12345-01。转换需要剥离-01。