从多个CSV中提取一列并保存到新CSV

zengzsys 于 2022-12-06 发布在其他

关注(0)|答案(2)|浏览(210)

我想从100多个CSV文件中读取特定列以创建新的CSV文件。源列的标题将被重命名为该列的提取文件名。
我可以获得各个列，但如果不使用.“csv”扩展名，则无法重命名每个列的标题：

import os
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)

csv

来源：https://stackoverflow.com/questions/70292225/extract-single-column-from-multiple-csvs-and-save-to-new-csv

2条答案

按热度按时间

eivnm1vs1#

这是最后一段对我有用的代码（参见输出1）：

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)
        E_1 = E.rename(columns={file: file.name.split('.')[0]}) # rename df header while dropping the ext **[.csv]** and the `os.scandir` attribute `<DirEntry>`
        E2080.append(E_1)

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

赞(0）回复(0）举报 2022-12-06

t9aqgxwy2#

您应该使用file.name而不是file来取得名称为的string。
而对于string，您可以使用.split(".")来获取不带扩展名的名称。

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

或者您可以使用pathlib.Path而不是os.scandir()来拥有更多的函数。

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

赞(0）回复(0）举报 2022-12-06

我来回答

从多个CSV中提取一列并保存到新CSV

2条答案

相关问题

热门标签

最新问答