【发布时间】:2017-10-11 12:01:50
【问题描述】:
我正在尝试根据从 csv 文件创建的列表将文件从一个目录复制到另一个目录。现在我有一个包含几列的 csv 文件,但我设法提取了我需要的一列并将其保存在“imaglst”下。
该列表是图像“.tif”扩展名的列表。我有一个输入文件夹(包含所有图像和额外的图像)和一个输出文件夹(我想在其中复制“imaglst”中指定的图像)。
以下是当前代码:
import os
import glob
import pandas
import shutil
cwd = os.getcwd()
path = cwd
extension = 'csv'
result = [i for i in glob.glob('*.{}'.format(extension))]
print result
colnames = ['X' ,'Y' ,'ATTR_1',' ATTR_2',' ATTR_3 ','ATTR_4' ,'ELEVATION']
data = pandas.read_csv(result[0],names=colnames, delimiter=r"\s+")
imaglst = data['ATTR_1']
print imaglst[1:len(imaglst)]
dir_src = raw_input('enter the full image folder location :')
dir_dst = raw_input('enter the full output folder location :')
for i in imaglst[1:len(imaglst)]:
for filename in imaglst.filter(os.listdir(dir_src), imaglst[i]):
shutil.copy(dir_src, dir_dst)
print "---------------------------------------------------------------------------------"
print "Process competed"
打印时保存的列表
打印 imaglst[1:len(imaglst)]
1 17251_0002_RGB
2 17251_0004_RGB
3 17251_0006_RGB
4 17251_0008_RGB
5 17251_0010_RGB
6 17251_0012_RGB
7 17251_0014_RGB
8 17251_0016_RGB
9 17251_0018_RGB
10 17251_0020_RGB
11 17251_0022_RGB
12 17251_0024_RGB
13 17251_0026_RGB
14 17251_0028_RGB
15 17251_0030_RGB
16 17251_0032_RGB
17 17251_0034_RGB
18 17251_0036_RGB
19 17251_0038_RGB
20 17251_0040_RGB
21 17251_0042_RGB
22 17251_0044_RGB
23 17251_0046_RGB
24 17251_0048_RGB
25 17251_0050_RGB
26 17251_0052_RGB
27 17251_0054_RGB
28 17251_0056_RGB
29 17251_0058_RGB
30 17251_0060_RGB
206 17005_0114_RGB
207 17005_0116_RGB
208 17005_0118_RGB
209 17005_0120_RGB
210 17005_0122_RGB
211 17005_0124_RGB
212 17005_0126_RGB
213 17005_0128_RGB
214 17005_0130_RGB
215 17005_0132_RGB
216 17005_0134_RGB
217 17005_0136_RGB
218 17005_0138_RGB
219 17005_0140_RGB
220 17005_0142_RGB
221 17005_0144_RGB
222 17005_0146_RGB
223 17005_0148_RGB
224 17005_0150_RGB
225 17005_0152_RGB
226 17005_0154_RGB
227 17005_0156_RGB
228 17005_0158_RGB
229 17005_0160_RGB
230 17005_0162_RGB
231 17005_0164_RGB
232 17005_0166_RGB
233 17005_0168_RGB
234 17005_0170_RGB
235 17005_0172_RGB
现在我知道我遗漏了一些东西,但不知道是什么,任何建议或解决方法将不胜感激。当我运行它时,出现以下错误:
KeyError: '17251_0002_RGB'
所以,我可以理解的是,我可能没有选择扩展名“.tif”,但我不确定。
【问题讨论】:
标签: python python-2.7 python-3.x csv