首页 > 解决方案 > 遍历 csv 文件中的值并创建目录

问题描述

我有一个结构如下的 csv 文件: 在此处输入图像描述

我需要遍历它的行并创建一个称为列值“worker_id”的文件夹,其中包含一个列值为“audio_id”的文件,其中有一个列值为“text”的文本。有我的代码:

import os
import pandas as pd

df = pd.read_csv('ya/audio_dataset/df.csv')
print(df.items())
print(df.head())
path0 = "1"

for text, audio_id, worker_id, task_id, path in df.items():
    path1 = path0 + worker_id
    if not (os.path.exists(path1)):
        try:
            os.mkdir(path1)
        except OSError:
            print("Создать директорию %s не удалось" % path1)
        else:
            print("Успешно создана директория %s " % path1)

    path2 = path1 + '/' + task_id

    if not (os.path.exists(path2)):
        try:
            os.mkdir(path2)
        except OSError:
            print("Создать директорию %s не удалось" % path2)
        else:
            print("Успешно создана директория %s " % path2)

    path3 = path2 + '/' + audio_id
    my_file = open("path3" + ".txt", "w+")
    my_file.write("text")
    my_file.close()

    os.rename('ya/audio-dataset/' + path, path3 + ".wav")

但我收到以下错误:

Traceback (most recent call last):
  File "D:/dip/main.py", line 9, in <module>
    for text, audio_id, worker_id, task_id, path in df.items():
ValueError: not enough values to unpack (expected 5, got 2)

标签: pythonpandas

解决方案


推荐阅读