Python代码：从多个csv文件中提取单个列以另存为新的csv文件，同时列头==源csv文件

Question

我想从多个（100秒）CSV文件中读取特定列，以创建新的CSV文件。列标题将使用提取列的源文件名重命名。我已经为此编写了一些代码行，但是，在没有扩展名的情况下，我无法有效地重命名列标题（请参阅附件）。我会感谢你的帮助。谢谢。

import os
import csv
import pandas as pd

folder = "C:/Users/Doc/Data"
files = os.scandir(folder)

E2080 = []
with os.scandir(folder) as files:
    for file in files:
        #print(file)
        df = pd.read_csv(file, index_col=None)
        dist = {file: (df['lnt_dist'])}
        E = pd.DataFrame(dist)
        E2080.append(E)

dist = pd.concat(E2080, ignore_index=False, axis=1)
dist.head()
dist.to_csv('E2080', index=False)

Answer 1

您应该使用file.name而不是file来获取带有名称的string。

使用string，您可以使用。拆分(".")获得没有扩展名的名称。

for file in os.scandir(folder):
    print(file.name, '=>', file.name.split(".")[0])

或者您可以使用pathlib。路径而不是操作系统。scandir（）以拥有更多功能。

for file in pathlib.Path('test').iterdir():
    print(file.name, '=>', file.stem)