我打开了许多 csv 文件，现在我需要将这些文件的名称添加到

Question

我在文件夹中有很多 csv 文件，它们的名称有 XCV123.csv XCF456.csv、DFT456.csv 等。我打开它们并保存为一个，但我发现其中没有这些文件的名称.csv。 DF 如下所示（当然一个 csv 有很多行）：

A B C 
1 2 3 
5 2 3
6 5 2

所以我要找的是像这样的 DF 或 CSV

A B C Name
1 2 3 XCV123
5 2 3 XCF456
6 5 2 DFT456

我用代码打开了这个：

import pandas as pd
from glob import glob

with open('/opt/home/main.csv', 'a') as singleFile:
    for csv in glob('*.csv'):
        if csv == 'main.csv':
            pass
        else:
            for line in open(csv, 'r'):
                singleFile.write(line)

但我必须在这段代码中添加一些内容，但我不知道是什么...感谢您的所有帮助！

最好的问候

Answer 1

我在这里假设：

• 所有 csv 文件都有相同的标题
• 所有 csv 文件都有相同的分隔符：逗号
• 您可以在 csv 文件中使用带引号的字段
• 您可以在每次运行时擦除文件

然后，您可以将代码更改为：

# use mode wb to control end of lines and overwrite the output file
with open('/opt/home/main.csv', 'wb') as singleFile:
    do_header = True                 # add 1! header
    for csv in glob('*.csv'):
        if csv == 'main.csv':
            pass
        else:
            with open(csv, 'rb') as file:
                line = next(file)
                if do_header:         # only on first pass
                    singlefile.write(line.rstrip() + b',Name\r\n')
                    do_header = False
                for line in file:
                    singlefile.write(line.rstrip() + ',"{}"\r\n'.format(csv).encode())
            singleFile.write(line)

Answer 2

使用pandas 和pathlib

您可以在 pathlib 对象上使用 .stem 以返回不带扩展名的文件名。

import pandas as pd
from pathlib import Path

files = Path('/to/csv/files').glob('*.csv') #rglob if you have sub directories.

dfs = pd.concat([pd.read_csv(file).assign(Name=file.stem) for file in files])

dfs.to_csv('singleFile.csv',index=False)