我正在嘗試在 python 中構建一個填充有 1 和 0 的數據框，具體取決于一列中的數字：Date Hour2005-01-01 12005-01-01 22005-01-01 32005-01-01 4我想根據“小時”中的數字創建新列，如果該行等于“小時”中的值，則用 1 填充每列，否則用 0 填充。Date Hour HE1 HE2 HE3 HE42005-01-01 1 1 0 0 02005-01-01 2 0 1 0 02005-01-01 3 0 0 1 02005-01-01 4 0 0 0 1我可以用這段代碼做到這一點，但需要很長時間：for x in range(1,5): _HE = 'HE' + str(x) for i in load.index: load.at[i, _HE] = 1 if load.at[i,'Hour']==x else 0我覺得這是 .apply() 的一個很好的應用程序（沒有雙關語），但我無法讓它正常工作。你會如何加快這個速度？
3 回答
海綿寶寶撒
TA貢獻1809條經驗 獲得超8個贊
pandas.factorize 和數組切片分配
j, h = pd.factorize(df.Hour)
i = np.arange(len(df))
b = np.zeros((len(df), len(h)), dtype=h.dtype)
b[i, j] = 1
df.join(pd.DataFrame(b, df.index, h).add_prefix('HE'))
Date Hour HE1 HE2 HE3 HE4
0 2005-01-01 1 1 0 0 0
1 2005-01-01 2 0 1 0 0
2 2005-01-01 3 0 0 1 0
3 2005-01-01 4 0 0 0 1
浮云間
TA貢獻1829條經驗 獲得超4個贊
即使它與@jezrael 的答案非常相似，但這也更好，（它只是使用.str訪問器get_dummies：
print(df.join(df['Hour'].astype(str).str.get_dummies().add_prefix('HE')))
輸出：
Date Hour HE1 HE2 HE3 HE4
0 2005-01-01 1 1 0 0 0
1 2005-01-01 2 0 1 0 0
2 2005-01-01 3 0 0 1 0
3 2005-01-01 4 0 0 0 1
添加回答
舉報
0/150
提交
取消