用python将一个excel表格中的数据生成一个字典然后用生成的字典去对比另一个excel表格中的数据找到相同的数据在输出一个新的excel表格
时间: 2024-03-19 11:44:08 浏览: 19
好的,我来解答你的问题。
首先,我们需要使用 `openpyxl` 模块来读取和写入 Excel 文件。你可以使用以下命令行安装这个模块:
```
pip install openpyxl
```
接下来,我们来看看如何将一个 Excel 表格中的数据生成一个字典。假设我们有一个名为 `data.xlsx` 的 Excel 文件,其中包含以下数据:
| 姓名 | 年龄 | 性别 |
| --- | --- | --- |
| 小明 | 20 | 男 |
| 小红 | 18 | 女 |
| 小刚 | 22 | 男 |
| 小美 | 19 | 女 |
我们可以使用以下 Python 代码将这些数据读取到一个字典中:
```python
from openpyxl import load_workbook
wb = load_workbook('data.xlsx')
ws = wb.active
data = {}
for row in ws.iter_rows(min_row=2, values_only=True):
name, age, gender = row
data[name] = {'age': age, 'gender': gender}
```
上面的代码中,我们首先使用 `load_workbook` 函数加载 Excel 文件。然后,我们使用 `active` 属性获取当前活动的工作表。接着,我们使用 `iter_rows` 方法遍历工作表中的每一行数据。由于第一行是标题,我们使用 `min_row=2` 参数跳过第一行。然后,我们使用 `values_only=True` 参数获取每一行的值,而不是单元格对象。最后,我们将每一行的数据存储到一个以姓名为键的字典中。
接下来,我们需要读取另一个 Excel 文件,并找到与第一个文件中相同的数据。假设我们有一个名为 `other.xlsx` 的 Excel 文件,其中包含以下数据:
| 姓名 | 地址 |
| --- | --- |
| 小明 | 北京市 |
| 小红 | 上海市 |
| 小刚 | 广州市 |
| 小丽 | 深圳市 |
我们可以使用以下 Python 代码将这些数据读取到一个新的字典中,并查找与第一个文件中相同的数据:
```python
other = {}
for row in ws.iter_rows(min_row=2, values_only=True):
name, address = row
other[name] = address
result = {}
for name, info in data.items():
if name in other:
result[name] = {'age': info['age'], 'gender': info['gender'], 'address': other[name]}
```
上面的代码中,我们首先使用 `iter_rows` 方法遍历另一个 Excel 文件中的每一行数据,并将其存储到一个以姓名为键的字典中。然后,我们遍历第一个 Excel 文件中的每一行数据。如果当前行的姓名在第二个文件中也存在,我们就将该行数据和第二个文件中对应的地址存储到一个新的字典中。
最后,我们需要将这个新的字典写入到一个 Excel 文件中。我们可以使用以下 Python 代码实现:
```python
from openpyxl import Workbook
wb = Workbook()
ws = wb.active
ws.append(['姓名', '年龄', '性别', '地址'])
for name, info in result.items():
ws.append([name, info['age'], info['gender'], info['address']])
wb.save('result.xlsx')
```
上面的代码中,我们首先创建一个新的 Excel 文件,并获取当前活动的工作表。然后,我们使用 `append` 方法将标题行写入工作表。接着,我们遍历新的字典中的每一行数据,将其写入到工作表中。最后,我们使用 `save` 方法将工作表保存到一个名为 `result.xlsx` 的文件中。
希望这样的回答对你有所帮助!