python - 如何使用熊猫从 csv 中的绝对路径导入数据？

Question

我有一个包含 n 条记录的 CSV，其中填充了图像的绝对路径。我想将这些图像导入一个 numpy 矩阵。

标签： pythonpandas

Answer 1

import pandas as pd
from PIL import Image
import numpy as np

def load_image( infilename ) :
  img = Image.open( infilename )
  img.load()
  data = np.asarray( img, dtype="int32" )
  return data

df = pd.read_csv (r'Path where the CSV file is stored\File name.csv')

for i in range(len(df)) : 
  print(load_image(df.iloc[i, 0]))

如果您想直接使用，可以将返回的值存储在列表中。