我有一個(gè)只有一個(gè)值的數(shù)據(jù)框，我想將它與數(shù)值進(jìn)行比較并根據(jù)它引發(fā)錯(cuò)誤。是這樣的：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() 打印它，但無法正確比較，因?yàn)橐粋€(gè)是數(shù)據(jù)幀，另一個(gè)是整數(shù)。請指教。
1 回答
SMILET
TA貢獻(xiàn)1796條經(jīng)驗(yàn) 獲得超4個(gè)贊
您需要收集要在 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!")
添加回答
舉報(bào)
0/150
提交
取消