python - Pandas 创建一个新列来说明文件是否存在

Question

从数据集中解析数据，其中某些图像不可用，因此我想创建一个新行 exists这样我就可以循环遍历图像名称 <id>.jpg放在那里“假”或“真”。

出现 unicode 错误

import pandas as pd
from pandas import Series
train = pd.read_csv('train.csv')

In [16]: train['exists'] = Series(str(os.path.isfile('training_images/' + train['id'] + '.jpg')))
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-16-4ada5144d198> in <module>()
----> 1 train['exists'] = Series(str(os.path.isfile('training_images/' + train['id'] + '.jpg')))
/usr/lib/python2.7/genericpath.pyc in isfile(path)
     35     """Test whether a path is a regular file"""
     36     try:
---> 37         st = os.stat(path)
     38     except os.error:
     39         return False
TypeError: coercing to Unicode: need string or buffer, Series found

Answer 1

我建议您使用矢量化解决方案，如下所示:

train['filename'] = 'training_images' + os.sep + train['id'] + '.jpg'
train['exists'] = train['filename'].map(os.path.isfile)

结果将是一个 bool 值pd.Series。