python - 使用 Pandas 和 Regex 将文件名写入 csv

Question

如何使用 Pandas 将文件名列表写入 csv 文件中的列？我还希望 Regex 只保留文件名的一部分。

对于 csv 模块，我是这样做的:

import os
import re
import csv

with open("file.csv","w") as write_csv:
    fieldnames = ["col1", "col2"]
    wr_header = csv.DictWriter(write_csv, fieldnames = fieldnames)
    wr_header.writeheader()
    wr_data = csv.writer(write_csv, delimiter = "\n")
    folder = os.listdir("/User/folder")
    for filename in folder:
        filesnippet = re.findall("snippet_(.{12})", filename)
        wr_data.writerow(filesnippet)

Answer 1

Pandas dataframe 可以将列表作为构造函数。真正的问题是你在哪里做正则表达式的工作。在这里，我使用 pandas 字符串方法 extract 返回每行捕获的组。

pd.Series(os.listdir(directory)).str.extract(regex).to_csv(filename)

参见 http://pandas.pydata.org/pandas-docs/version/0.15.2/generated/pandas.core.strings.StringMethods.extract.html