Python For循環/列表理解復制多列

Question

我覺得有點愚蠢，但我已經好幾個小時沒有得到這個了。 我有一個帶有兩列的 Pandas DataFrame。 現在，我想遍歷“firstcol”列並檢查一個條件（這里：字符串的長度等等）。 結果應該被復制到一個新的列表/df。

result = [i for i in data["firstcol"] if (len(i)>5 and ...)]

問題：現在我錯過了第二列。 但我想復制整行。 我怎樣才能做到這一點？

謝謝！

Answer 1

您不需要循環整個 dataframe，而是可以使用 apply 方法並將 function 應用於整個列。

嘗試這樣的事情：

df_new = df[df['firstcol'].apply(lambda x: len(str(x)) > 5)]