python - 用python从pc中的路径中提取多个csv文件信息并同时操作

Question

我的文件夹路径D:\sevenday中有多个（比如说超过 20/30）名为'ABCD.csv'、'EFGH.csv'、'IJKL.csv'、'MNOP.csv' 等的CSV 文件。我想在每个 CSV 文件中添加一个名为name的列，并将它们各自的名称添加到 CSV 文件本身中。让我们说，ABCD CSV 文件将有一个名为name的列，并且它将ABCD作为该列中的名称。如何从 python 中的文件名中提取该字符串信息？有办法吗？如果文件很少，有办法。但是我不知道如何一次提取许多这样的文件，因为每次一个一个地编写很乏味。

标签： pythonpandasdataframecsvdata-manipulation

Answer 1

用于glob从您的文件夹中获取所有文件以列出，添加新列并回写：

import glob, os

files = glob.glob('D:/sevenday/*.csv')

for fp in files:
    name = os.path.basename(fp)
    df = pd.read_csv(fp)
    df['name'] = name
    df.to_csv(fp, index=False)