如何看條件為真/假DataFrame中所有行嗎?

[麥當娜]

假設我有一個火花DataFrame,我想看看記錄滿足條件。

示例數據集:

#數據準備數據= [(' A ', 1), \ (' A ', 2) \ (“B”, 3)] #創建DataFrame列= [‘col_1’,‘col_2] df =火花。createDataFrame(=數據、模式=列)df.show(截斷= False)

如果我運行下麵的代碼在磚:

在輸出中,我沒有看到如果條件滿足。如果我創建了一個熊貓DataFrame:

熊貓作為pd導入pdf = pd。DataFrame(數據列=列)

我可以檢查,如果條件滿足所有行:

我怎麼能得到相同的輸出使用火花DataFrame時?

[UmaMahesh1 · Accepted solution]

嗨@Mohammad軍刀

因為你的輸出將一列對象你隻需要使用df。

下麵的工作。

df。選擇(df [' col_1 '] = = A),告訴()

希望這有助於……其標記為答案如果它幫助. .

幹杯

[麥當娜]

謝謝。我怎麼能改變列名”(col_1 =)”。“條件”?

[UmaMahesh1]

您可以使用一個“否則”,給你想要的條件。它的工作原理類似於sql查詢情況。

如。

df。選擇(當(df [' col_1 '] = = ' A ',“Condition1”) .otherwise (“Condition2”)),告訴()

[麥當娜]

謝謝。我的意思是改變列名。我試著:

df。選擇(df [' col_1 '] = = A) .alias(的條件),告訴()

但是,它沒有工作。