可以使用if语句来检查dataframe是否为空,如果为空则跳过对count()的调用。 代码示例如下:
'''在加载数据后创建一个空的dataframe''' df = spark.read.format("csv").option("header", True).load(csv_file) empty_df = df.filter("1=0")
'''检查dataframe是否为空,并避免对空dataframe调用count()方法''' if empty_df.rdd.isEmpty(): print("Empty dataframe!") else: print("Dataframe count:", empty_df.count())