使用Excel工作表中的数据替换数据框中的值

时间:2019-06-25 11:17:54

标签: python excel python-3.x pandas dataframe

我有一个带有产品名称的简单excel文件。第一行是类别(A1:水,A2:起泡,A3:仍然,B1:软饮料,B2:可口可乐,B3:橙汁,B4:柠檬水等),下面的每个单元格都是不同的产品。我想用产品类别替换另一个数据框(df1)的单元格。例如,可口可乐将成为软饮料。如果该产品不在excel中,则不会被替换(例如Cookie)。

A1: Water, A2: Sparkling, A3:Still, B1: Soft Drinks, B2: Coca Cola, B3: Orange Juice, B4:Lemonade etc.

print(df1)

         Product  Quantity
0      Coca Cola  1234
1      Cookie     4
2      Still      333
3      Chips     88

结果:

print (df1)

       Product      Quantity
0      Soft Drinks   1234
1      Cookie        4
2      Water         333
3      Snacks        88

1 个答案:

答案 0 :(得分:0)

我们从Excel数据框中创建一个查找字典,并将其用于用类别替换产品名称(在字典中找不到的产品将按要求保留不变):

df = pd.read_excel(r'c:\temp\product_names.xlsx')
df1 = pd.DataFrame({'Product': ['Coca Cola','Cookie','Still','Chips'], 'Quantity': [1234,4,333,88]})

lu = df.unstack().dropna().reset_index()
df1.Product.replace(dict(zip(lu.iloc[:,2],lu.iloc[:,0])),inplace=True)

结果:

       Product  Quantity
0  Soft Drinks      1234
1       Cookie         4
2        Water       333
3       Snacks        88

PS:这是csv文件,粘贴到A1处的Excel工作表中以重现示例:

Water,Soft Drinks,Sandwich,Snacks
Sparkling,Coca Cola,Ham and cheese,Chips
Still,Orange Juice,Vegetarian,
,Lemonade,Club,
,,Spinach,