获取int（）函数的参数必须是字符串或数字，而不是“Column'-阿帕奇星火

Question

如果我用下面的代码我得到这个异常：

int() argument must be a string or a number, not 'Column' 

df= df.withColumn('FY', 
    F.when((df['ID'].substr(5,2).isin({'11','12'})),int(df['ID'].substr(1,4))+1). 
    otherwise(int(df['ID'].substr(1,4)))) 

基本上如果结果是11或12，我想给结果加1，否则就是ID。请帮助，我对Python相当陌生。

Answer 1

用途：

df.withColumn('FY',F.when(df['ID'].substr(5,2).isin({'11','12'}), 
    df['ID'].substr(1,4).cast("integer") + 1). 
    otherwise(df['ID'].substr(1,4)).cast("integer"))