PYTHON 遍历两个文件夹,第一个文件夹储存900个物体的位置信息,第二个文件夹储存900个物体的形状信息,现在要先两两比较第一个文件夹内物体的位置信息,如果一样的话在比较他们的形状信息
时间: 2024-05-03 13:22:45 浏览: 44
,如果形状信息也一样,就将这两个物体标记为相同的物体。
以下是示例代码:
```python
import os
# 定义两个文件夹的路径
pos_folder = "path/to/position/folder"
shape_folder = "path/to/shape/folder"
# 获取两个文件夹中的文件列表
pos_files = os.listdir(pos_folder)
shape_files = os.listdir(shape_folder)
# 遍历第一个文件夹
for pos_file in pos_files:
# 获取物体编号
obj_id = pos_file.split(".")[0]
# 打开位置信息文件
with open(os.path.join(pos_folder, pos_file), "r") as pos_f:
pos_data = pos_f.read()
# 遍历第二个文件夹
for shape_file in shape_files:
# 判断文件名是否匹配
if shape_file.startswith(obj_id):
# 打开形状信息文件
with open(os.path.join(shape_folder, shape_file), "r") as shape_f:
shape_data = shape_f.read()
# 比较位置信息和形状信息是否一致
if pos_data == shape_data:
print(f"物体 {obj_id} 的位置信息和形状信息相同,这是同一个物体。")
```
注意,上述代码假设位置信息和形状信息是以文本文件的形式存储的,如果是其他格式的文件,需要使用相应的读取方式。另外,比较字符串是否相同时,需要注意换行符等细节。
阅读全文