Question

文件夹中有很多csv excel工作表。所有Excel工作表仅在前3列中包含数据。我将从大量的csv工作表中选择相应的csv工作表，然后将其绘制。这是代码

import os
path = "F:\\Users\\Desktop\\Data\\Summary"
files = []
folder_data = os.listdir(path)
folder_data = [i+"\\" for i in folder_data]

# r=root, d=directories, f = files
for r, d, f in os.walk(path):
    for file in f:
        if '.csv' in file:
            files.append(file)
for i, f in enumerate(files):
    print(( i,f))
    print('\n'.join(f'{i}-{v}' for i,v in enumerate(files)))
    csv_code = str(int(input("Enter corresponding code to plot: ")))
    csv_path = path + "\\" + folder_data[csv_code] 

    df = pd.read_csv(csv_path, header=None)
    df1 = df.iloc[:,0:2]
    plt.plot(df1[0], df1[1])

当我运行代码时，我希望输出显示如下（我的意思是我希望显示该文件夹中的所有csv文件，以便我可以选择想要的内容）：

0-Test_Summary_1.csv
1-Test_Summary_2.csv
2-Test_Summary_3.csv
3-Test_Summary_4.csv
4-Test_Summary_5.csv
5-Test_Summary_6.csv etc

我得到的错误是

FileNotFoundError:

尽管该文件夹中有csv文件，但该文件仍然存在。我收到“找不到文件错误”错误

Answer 1

如果我正确理解了您的问题，那么您可以尝试执行以下操作：

import os
import pandas as pd

# see this answer about absolute paths in windows
# https://stackoverflow.com/a/7767925/9225671
base_path = os.path.join('f:', os.sep, 'Users', 'Desktop', 'Data', 'Summary')

# collect all CSV files in 'base_path' and its subfolders
csv_file_list = []
for dir_path, _, file_name_list in os.walk(base_path):
    for file_name in file_name_list:
        if file_name.endswith('.csv'):
            # add full path to the list, not just 'file_name'
            csv_file_list.append(
                os.path.join(dir_path, file_name))

print('CSV files that were found:')
for i, file_path in enumerate(csv_file_list):
    print('   {:3d} {}'.format(i, file_path))

selected_i = int(input('Enter corresponding number of the file to plot: '))
selected_file_path = csv_file_list[selected_i]
print('selected_file_path:', selected_file_path)

df = pd.read_csv(selected_file_path, header=None)
...

这对您有用吗？

FileNotFoundError：

1 个答案: