我有一個只有一個值的數據框,我想將它與數值進行比較并根據它引發錯誤。是這樣的:df=spark.sql("""select count(1) as count_check from schema.table) """)df.select("count_check").show()if df.value == 0: print("ALL GOOD")else: raise Exception("ATAR DATA MISMATCH, RESULT COUNT IS NOT ZERO!")我可以使用 show() 打印它,但無法正確比較,因為一個是數據幀,另一個是整數。請指教。
1 回答

SMILET
TA貢獻1796條經驗 獲得超4個贊
您需要收集要在 python 上下文中使用的 Dataframe 列值
df = spark.sql('select count(1) as count_check from schema.table')
value = df.collect()[0][0]
if value == 0:
print("ALL GOOD")
else:
raise Exception("ATAR DATA MISMATCH, RESULT COUNT IS NOT ZERO!")
添加回答
舉報
0/150
提交
取消