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 *