Azure Databricks 和 pyspark - 子串错误

Question

使用以下行在我的 Databricks Spark 脚本中出现两个错误：

df = spark.createDataFrame(pdDf).withColumn('month', substring(col('dt'), 0, 7))

第一个：

AttributeError: 'Series' object has no attribute 'substr'

和

NameError: name 'substr' is not defined

我想知道我做错了什么...

Answer 1

原来我没有导入pyspark.sql.functions

from pyspark.sql.functions import *