如何将数据从 Excel 输入到 Python 字典？

Question

我的代码：

import openpyxl
workbook = openpyxl.load_workbook('master.xlsx')
worksheet = workbook.worksheets[0]
result = {}
for k, v in zip(worksheet['A'], worksheet['B']):
    result[k.internal_value] = v.internal_value
 print(result)

output 我得到：

{'PPPPP': '22', 'bbbbb': '20', 'ccccc': '30', 'ddddd': '40', 'eeeee': '50'}

Excel 文件：

我想要的 output：

{'PPPPP': ['22','10'], 'bbbbb': ['20','30'], 'ccccc': ['30','30'], 'ddddd': '40', 'eeeee': '50'}

Answer 1

使用一个 defaultdict，一个空列表作为默认值，append 每个新值：

from collections import defauldict
import openpyxl
workbook = openpyxl.load_workbook('master.xlsx')
worksheet = workbook.worksheets[0]
result = defaultdict(list)
for k, v in zip(worksheet['A'], worksheet['B']):
    result[k.internal_value].append(v.internal_value)
 print(result)

在这里，每个结果都将是一个列表，即使您只有一个值。 例如，您将得到'ddddd': ['40']但您应该能够一致地处理所有键值对。

Answer 2

您可以使用pandas

import pandas as pd
df = pd.read_excel('master.xlsx', 0, None, ['A', 'B'])
result = {}
for x, y in zip(df['A'], df['B']):
    if x in result:
        result[x] = [result.get(x)]
        result[x].append(str(y))
    else:
        result[x] = str(y)

print(result)
{'ppp': ['10', '22'], 'bbb': ['20', '30'], 'ccc': ['30', '30'], 'ddd': '40', 'eee': '50'}