用Python将deepfashion数据集的list_attr_img.txt变成CSV文件,不使用pandas
时间: 2024-03-24 18:40:41 浏览: 16
如果您不想使用pandas库,也可以使用Python将DeepFashion数据集中的list_attr_img.txt文件转换为CSV文件。可以按照以下代码进行操作:
```python
# 读取list_attr_img.txt文件
with open('list_attr_img.txt', 'r') as f:
lines = f.readlines()
# 处理每行数据
data = []
for line in lines[2:]:
line = line.strip().split()
img_path = line[0]
attrs = line[1:]
attrs = [int(x) for x in attrs]
data.append([img_path] + attrs)
# 将数据保存为CSV文件
with open('list_attr_img.csv', 'w') as f:
# 写入列名
f.write('image_name,')
f.write(','.join(lines[1].strip().split()[1:]) + '\n')
# 写入数据
for line in data:
line = [str(x) for x in line]
f.write(','.join(line) + '\n')
```
解释:
1. 首先使用open()函数读取list_attr_img.txt文件,将每行数据读取为一个字符串。
2. 对于每行数据,使用strip()函数去除空格和换行符,并使用split()函数将其拆分为图像路径和属性值。
3. 将属性值转换为整数类型。
4. 将图像路径和属性值组成一个列表,并添加到data列表中。
5. 使用open()函数创建CSV文件,并使用write()函数将data列表中的数据写入CSV文件中。在写入CSV文件时,首先使用join()函数将列名转换为一个字符串,然后写入CSV文件;接着,使用join()函数将每行数据转换为一个字符串,并使用逗号分隔每个数据项。
这样,就可以使用Python将DeepFashion数据集中的list_attr_img.txt文件转换为CSV文件了。