根据情节选择复制文件

Question

我在一个目录中有许多 .txt 文件。起初我想在屏幕上一个一个地绘制文件，如果看起来不错，那么我想将 .txt 文件复制到一个名为“test_folder”的目录中。 如果它看起来不正常，那么我不想将 .txt 文件复制到“test_folder”目录。

我尝试了下面的脚本，但是我不能这样做，因为我是 python 新手。我希望专家可以帮助我克服这个问题。在此先感谢。

import numpy as np
import os,glob,shutil
import matplotlib.pyplot as plt

os.mkdir('test_folder')

for filex in glob.glob("*.txt"):
    print(filex)
    data=np.loadtxt(filex)
    plt.plot(data)
    plt.show()

    if plot_looks_nice == "yes": 
    #copy the filex to the directory "test_folder"
       shutil.copy(filex,'test_folder')
    elif plot_looks_nice == "no": 
    #donot copy the filex to the directory "test_folder"
         print("not copying files as no option chosen")
    else: 
    print("Please enter yes or no.") 

Answer 1

你很接近。您想使用input() 来提示用户并通过他们的输入返回一个变量。

创建目录的最佳方法是使用 pathlib (python >= 3.5) 递归地创建不存在的目录。这样您就不必担心由于目录不存在而导致的错误

查看下面的修改代码。

import numpy as np
import os,glob,shutil
import matplotlib.pyplot as plt
from pathlib import Path

Path("test_folder").mkdir(exist_ok=True)

for filex in glob.glob("*.txt"):
    print(filex)
    data=np.loadtxt(filex)
    plt.plot(data)
    plt.show()

    plot_looks_nice = input('Looks good? ')

    if plot_looks_nice == "y": #use single letters to make your work faster
    #copy the filex to the directory "test_folder"
       shutil.copy(filex,'test_folder')
    elif plot_looks_nice == "n": 
    #donot copy the filex to the directory "test_folder"
         print("not copying files as no option chosen")
    else: 
    print("Please enter \'y\' or \'n\'.")