从一个数据帧值与一个数据帧列合并

Question

我这里有一个困难。我的目标是为一个商店创建一个销售列表，其中一个数据框按产品列出价格，另一个数据框按产品和数量列出所有销售（一段时间内）

数据帧1：价格

prices = pd.DataFrame({'qty_from' : ('0','10','20'), 'qty_to' : ('9','19','29'), 'product_A' :(50,30,10),'product_B' :(24,14,12),'product_C' :(70,50,18)})

数据帧2：销售

sales = pd.DataFrame({'product' : ('product_b','product_b','product_a',product_c,product_b), 'qty' : ('4','12','21','41','7')})

我想得到营业额，在'销售'数据框内逐行，另一列像'营业额'

我曾经

pd.merge_asof(sales, prices, left_on='qty', right_on='qty_from', direction='backward') 

它给了我一个正确的价格，但如何得到一个产品的好价格呢？ 如何与“sales”数据框中的一个值（如“product\u b”）合并，并在“product\u b”中使用数据框价格中的一列名称，然后应用计算得到营业额

谢谢你的帮助

埃里克

Answer 1

如果我理解正确，您可以修改数据帧价格，以便能够使用merge_asof中的参数by，使用stack：

#modify price
prices_stack = (prices.set_index(['qty_from','qty_to']).stack() # then products become as a column
                     .reset_index(name='price').rename(columns={'level_2':'product'}))

# uniform the case
sales['product'] = sales['product'].str.lower()
prices_stack['product'] = prices_stack['product'].str.lower()
# this is necessary with your data here as not int
sales.qty = sales.qty.astype(int)
prices_stack.qty_from = prices_stack.qty_from.astype(int)

#now you can merge_asof adding by parameter
sales_prices = (pd.merge_asof( sales.sort_values('qty'), prices_stack, 
                               left_on='qty', right_on='qty_from', 
                               by = 'product', #first merge on the column product
                               direction='backward')
                  .drop(['qty_from','qty_to'], axis=1)) #not necessary columns

print (sales_prices)
     product  qty  price
0  product_b    4     24
1  product_b    7     24
2  product_b   12     14
3  product_a   21     10
4  product_c   41     18