嘗試使用df.idxmax時內存不足

Question

我有一個有49700行和790列的數據框。我試圖得到每行有什麼最大值的列。以下是我的代碼：

df['Max'] = df.idxmax(axis = 1) 

我收到內存錯誤。有沒有辦法解決這個問題。

Answer 1

你在32位系統上嗎？我在32位系統上也有數據幀操作的內存問題。數據框包含一個〜350mb的csv文件。

我的腳本使用多個數據框，所以我的解決方法是在此數據框操作之前從內存中刪除一些其他數據框。 我通過將另一個數據幀設置爲「None」來完成它。