目前正在嘗試使用 Pandas 和 Matplotlib 收集有關數據科學的一些知識。我想做什么選擇具有相應價格的鄰里團體租用私人房間（AirBnB）問題我得到一列的正確數據，但其余的我收到 NaN 值。感謝任何幫助，您可以在下面找到代碼片段和輸出。 ap_df = pd.DataFrame()for neighbourhood_group in df['neighbourhood_group'].unique(): nbhg_df = df.copy()[df['neighbourhood_group']==neighbourhood_group] nbhg_df[f"{neighbourhood_group}_Price"] = nbhg_df['price'] nbhg_df.set_index("id", inplace=True) nbhg_df.sort_index(inplace=True) if ap_df.empty: ap_df = nbhg_df[[f"{neighbourhood_group}_Price"]] else: ap_df = ap_df.join(nbhg_df[f"{neighbourhood_group}_Price"])ap_df
1 回答
米琪卡哇伊
TA貢獻1998條經驗 獲得超6個贊
所以我想我可以用另一個更簡單、更有效的例子來幫助你滿足你的需要
import pandas
df = pandas.DataFrame({
'id': [1, 2, 3],
'cat': ['cat_1', 'cat_2', 'cat_3'],
'price': [1, 2, 3]
})
df.set_index(df['id'], drop=True, inplace=True)
new_df = pandas.DataFrame(index=df.index)
for var in df['cat'].unique():
new_df[f'{var}_price'] = df[df['cat'] == var]['price']
new_df
輸出 ：
cat_1_price cat_2_price cat_3_price
id
1 1.0 NaN NaN
2 NaN 2.0 NaN
3 NaN NaN 3.0
添加回答
舉報
0/150
提交
取消