python - 将 csv 目录读入 Pandas Dataframe

Question

我有一个包含 csvs 的目录，其文件名代表数据库行的 ID。

我想将此目录读入 Pandas 数据框并将其加入现有数据框。

Python 中是否有任何方法可以将“ls”命令的结果读入 pandas Dataframe？

我已经尝试使用下面的代码获取一串文件名，但我无法弄清楚之后如何将它放入数据框中。

import os

files = ''
for root, dirs, files in os.walk("."):
    for filename in files:
        files += filename

Answer 1

您可以遍历文件，现在您只需要读取 csv 并将它们连接到数据框上。

import os

import pandas as pd


df = None

for root, dirs, files in os.walk('.'):
    for filename in files:
        if not df:
            df = pd.read_csv(filename)
            df['filename'] = filename
            continue
        tmp = pd.read_csv(filename)
        tmp['filename'] = filename
        df = pd.concat(df, tmp)