我正在尝试根据从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"
我打印时保存的列表
print 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",但我不确定。
答案 0 :(得分:1)
问题出在你的for循环中。
Python循环真正适用于每个循环;在每次迭代中,i
是imaglst
的实际值。您尝试将其作为计数器用于索引回列表,但您只需要使用该值。
此外,之后您的shutil.copy
电话不会引用文件名。我希望你的意思是用文件名加入dir_src。
最后,在切片列表时,如果只是长度,则不需要包含端点。
所以,把它们放在一起:
for i in imaglst[1:]:
for filename in imaglst.filter(os.listdir(dir_src), i):
shutil.copy(os.path.join(dir_src, filename), dir_dest)