使用pandas读取csv文件中的目录

Question

我有一个csv文件，如下所示

/user/desktop/1.jpg, 0
/user/desktop/2.jpg, 0

第一列是文件目录，第二列是类标签

我想要的是逐个阅读目录，我该怎么做？

现在我尝试的是读取csv文件并指定列名并读取列但它似乎无法正常工作

df = pandas.read_csv('csv_file', names=['name','class'])
x = matplotlib.image.imread(df['name'])

Answer 1

我认为你不应该像这样使用熊猫。 更好的是打开文件并逐行读取，如：

import csv    
with open('file.csv', 'r') as f:
    reader = csv.reader(f)
    for row in reader:
         img_path = row[0]
         img_class = row[1]
         matplotlib.image.imread(img_path)