根据从csv文件创建的列表将文件从一个目录复制到另一个目录

时间:2017-05-12 08:38:06

标签: python python-2.7 python-3.x csv

我正在尝试根据从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",但我不确定。

1 个答案:

答案 0 :(得分:1)

问题出在你的for循环中。

Python循环真正适用于每个循环;在每次迭代中,iimaglst的实际值。您尝试将其作为计数器用于索引回列表,但您只需要使用该值。

此外,之后您的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)