如何从excel链接熊猫中的分层数据？

Question

我有这个 excel 表，其层次结构如下： excel 片段

Item Category Price
**Electronics** 1 
Laptop 1 1000
Kindle 1 200
Mobile 1 500
**HouseItems** 2
VacuumCleaner 2 200
Clock 2 50

我如何按类别获取项目？ 例如，获取笔记本电脑等电子产品及其价格，并在单独的列表中获取家居用品。 在excel表中我有更多的类别，这只是一个片段。

df = pd.read_excel('items.xlsx',
                   ['itemSheet'], engine='openpyxl')
df['items'] = pd.Series()
item_list= ['Electronics', 'HouseItems']
for item in df['itemSheet']['Item']:
    if item in cost_entry_group:
        df['items'].add(item)

print(df['items'])

我如何将 itemCategory(electronics) 链接到笔记本电脑、kindle 和手机以及它们各自的价格，并对家居用品做同样的事情？

Answer 1

这个类别不是已经在你的 df 中了吗？

使用df[df['Category'] == 1]来获取类别等于 1 又名“HouseItems”的项目

您还可以执行以下操作：

categories = {'Electronics': 0, 'HouseItems': 1}
dfs  = {}
for category_name, category_number in categories.items():
     dfs[category_name] = df[df['Category'] == category_number]

获取仅包含一个类别的多个 DataFrame。

要从 DataFrame 中提取类别，您可以检查价格中的“nan”值：

categories = {}
for index, row in df.iterrows():
    if row.isnull().values.any():
        categories[row['Item']] = row['Category']

这将是我快速而肮脏的解决方案。 或者，您可以使用 openpyxl 浏览 Excel 表并检查粗体文本：

from openpyxl import load_workbook

wb = load_workbook(path, data_only=True)
sh = wb[wb.sheetnames[0]]
categories = {}
for i in range(sh.min_row, sh.max_row): # go through rows
    if sh['A{}'.format(i)].font.b == True: # font.b gives True if bold otherwise False
        name = sh['A{}'.format(i)].value
        number = sh['B{}'.format(i)].value
        categories[name] = number

可能有“更好”的解决方案，但它有效。