我有一個 pandas 資料框，它有一些空值，并且想要添加一個新列model_prediction，它是模型對資料的預測。我擁有的模型不采用空值，我希望model_prediction這些行的值為 NaN。問題是資料框非常大，使用 df.iterrows 是一個非常緩慢的程序，希望避免它。
uj5u.com熱心網友回復：
假設您的資料框是df并且模型是model，請試試這個：
import numpy as np
df = df.reset_index(drop=True)
df_na = df[df.isna().any(axis=1)]
df_na.loc[:,'model_prediction'] = np.nan
df_model = df.dropna()
df_model.loc[:,'model_prediction'] = model.predict(df_model.values)
df = df_model.append(df_na).sort_index()
轉載請註明出處，本文鏈接：https://www.uj5u.com/shujuku/444014.html
標籤：熊猫
上一篇：用Python去除例外值