在 python 中創建新的數據框列和填充值的有效方法是什么？

Question

我有一對列"car_model"和"year" ，我需要將它們作為tuple發送到函數，它會返回一個價格（浮動）。

如何迭代數據幀行，將"car_model"和"year"值發送到函數並將返回值添加到新列"price" ？

我在想：

model_year = CAR[["car_model", "year"]]

for x in model_year.to_numpy():
    model_year_tuple = tuple(x)
    price = calculate_price(model_year_tuple)
    //how to add to the column? the line below will always use the last calculated price
    CAR['price'] = price

Answer 1

我們可以做的

model_year['out'] = model_year.agg(tuple,1).map(calculate_price)

Answer 2

嘗試apply ：

CAR['price'] = model_year.apply(lambda x: calculate_price(tuple(x)), axis=1)

或列表理解：

CAR['price'] = [calculate_price(x) for x in zip(CAR['car_model'], CAR['year'])]

也就是說，您應該嘗試重寫您的calculate_price函數，以便它接受 numpy 數組而不是普通的 Python 元組。

Answer 3

這應該有效

 df['price'] = df.apply(lambda x: price((x['car_model'],x['year'])))