如何在Databricks Notebook中无表数据时终止执行？

下面是几种实用的实现方式，能精准判断查询结果是否为空，并终止Notebook执行：

方法1：使用DataFrame的isEmpty属性（推荐） #

这是Spark提供的原生属性，内部会做优化，无需全表扫描，效率最高：

df = sql("select * from TableA")
if df.isEmpty:
    dbutils.notebook.exit("表TableA无数据，终止执行")

方法2：通过count()判断（适合小表） #

如果是数据量不大的表，可以直接统计行数，等于0则终止：

df = sql("select * from TableA")
if df.count() == 0:
    dbutils.notebook.exit("表TableA无数据，终止执行")

注意：这个方法会扫描全表，大表场景下性能较差，不推荐。

方法3：通过limit(1)检查是否有数据 #

和isEmpty逻辑类似，只尝试获取一行数据，避免全表扫描：

df = sql("select * from TableA")
# collect()返回空列表说明无数据
if not df.limit(1).collect():
    dbutils.notebook.exit("表TableA无数据，终止执行")

你之前看到的df.TableA[0]==0写法是错误的，DataFrame并没有这种按表名直接索引的语法，所以无法生效。

内容的提问来源于stack exchange，提问作者Patterson