[已解决]求助关于python操作excel的用法

xue11

现在有6种水果，分别含有不同的元素，有一种或者几种元素，如下图0119001


现在已经用openpyxl将所有元素提出并去重，但是有些元素含有几种水果，想实现如下面的统计（如图0119002），已经尝试很多次没有找出解决方法；


特求助，使用openpyxl或者pandas 实现均可，谢谢各位。

最佳答案

月排行榜 / 总排行榜

XiaoPaiShen

2020-1-19 14:56:24

import pandas as pd
import easygui as g

excel_file = g.fileopenbox(msg=None, title=None, default="*.xlsx", filetypes=None, multiple=False)
df = pd.read_excel(excel_file, sheet_name=0)

result = {}
cols = ['类型', '含有元素']
df_list = df[cols]

for index, row in df_list.iterrows():
    fruit = row['类型']
    elements = row['含有元素'].split(',')

    for elem in elements:
        result[elem] = result.get(elem, list())
        result[elem].append(fruit)

# intialise data of lists. 
data = {'元素':[], '水果':[]} 
for key, value in result.items():
    data['元素'].append(key)
    data['水果'].append(','.join(value))

new_df = pd.DataFrame(data)

writer = pd.ExcelWriter(excel_file, engine='xlsxwriter')
df.to_excel(writer, sheet_name='Sheet1', index=False)
new_df.to_excel(writer, sheet_name='Sheet2', index=False)
writer.save() 

跳转到最佳答案楼层

XiaoPaiShen

import pandas as pd
import easygui as g

excel_file = g.fileopenbox(msg=None, title=None, default="*.xlsx", filetypes=None, multiple=False)
df = pd.read_excel(excel_file, sheet_name=0)

result = {}
cols = ['类型', '含有元素']
df_list = df[cols]

for index, row in df_list.iterrows():
    fruit = row['类型']
    elements = row['含有元素'].split(',')

    for elem in elements:
        result[elem] = result.get(elem, list())
        result[elem].append(fruit)

# intialise data of lists. 
data = {'元素':[], '水果':[]} 
for key, value in result.items():
    data['元素'].append(key)
    data['水果'].append(','.join(value))

new_df = pd.DataFrame(data)

writer = pd.ExcelWriter(excel_file, engine='xlsxwriter')
df.to_excel(writer, sheet_name='Sheet1', index=False)
new_df.to_excel(writer, sheet_name='Sheet2', index=False)
writer.save() 

EthanHsiung

我的想法是输出一个字典，key为元素，value为水果（列表）

import openpyxl
wb = openpyxl.load_workbook("example.xlsx")
dic = {}
ws = wb['Sheet1']
max_row =ws.max_row
for row in range(2,max_row+1):
    elements = ws['B'+str(row)].value
    element_list =list(elements.replace(',',''))
    #因为元素中有逗号，先将逗号用空字符替换，便于转换成列表
    for element in element_list:
        if element in dic.keys():
            dic[element].append(ws['A'+str(row)].value)
        else:
            dic[element]=[]
            dic[element].append(ws['A'+str(row)].value)
print(dic)

结果
{'a': ['苹果', '橘子', '山楂'], 't': ['香蕉', '橘子'], 'g': ['橘子', '梨'], 'b': ['橘子', '梨', '火龙果'], 'y': ['梨'], 'h': ['火龙果', '山楂'], 'n': ['山楂']}

wp231957

EthanHsiung 发表于 2020-1-20 11:20
我的想法是输出一个字典，key为元素，value为水果（列表）
结果
{'a': ['苹果', '橘子', '山楂'], 't': [ ...

这不是已经实现？？？

大头目

用0和1表示吧，例如：
        a        b        c
苹果        1        0        0
香蕉        1        1        0
橘子        0        0        1

彩虹七号

XiaoPaiShen 发表于 2020-1-20 12:48

这个写的漂亮