在多个 csv 文件中插入和重命名列并合并为一个 csv

Question

我有很多这样的 csv：

我想重命名列名，因为它们太长，并插入名为 Company 且具有相同值的新列（例如，aapl（Apple 的股票代码））。这是我如何做到这一点的代码：

data = pd.read_csv('aapl.csv', index_col=0)
data.insert(0, 'Company', 'aapl')
data.rename(columns={'Unnamed: 0': 'Company', 'Working Capital / Total Assets':'WC/TA', 'Retained Earnings / Total Assets': 'RE/TA', 'EBIT / Total Assets':'EB/TA','Current Market Cap / Total Liabilites':'MC/TL', 'Revenue / Total Assets':'RV/TA','Net Income / Total Assets (ROA)':'NI/TA', 'Total Liabilites / Total Assets':'TL/TA', 'Current Assets / Current Liabilites':'CA/CL'})
data.to_csv('aapl.csv')

我得到了

问题是，我有 278 家其他公司采用相同的 csv 格式（imb.csv、ase.csv、wire.csv 等），我想为所有公司重命名/插入列以同样的方式。最后，我需要将所有这些 csv 文件合并到一个 csv 中（这就是需要新列“公司”的原因）。

我的问题是，如何创建新列（名称：公司）并用文件名填充它。

例如：加载 ibm.csv -> 重命名列 -> 添加新列（公司）-> 插入值 ibm（文件名/股票代码）列公司 -> 导出到 csv。

Answer 1

您可以在字典理解中读取每个文件，按文件名添加键和最后concat：

import glob, os

files = glob.glob('files/*.csv')

d = {'Working Capital / Total Assets':'WC/TA', 
     'Retained Earnings / Total Assets': 'RE/TA', 
     'EBIT / Total Assets':'EB/TA',
     'Current Market Cap / Total Liabilites':'MC/TL',
     'Revenue / Total Assets':'RV/TA',
     'Net Income / Total Assets (ROA)':'NI/TA',
     'Total Liabilites / Total Assets':'TL/TA',
     'Current Assets / Current Liabilites':'CA/CL'}

dfs = {os.path.basename(fp).split('.')[0]: pd.read_csv(fp).rename(columns=d) for fp in files}
df = pd.concat(dfs).reset_index(level=1,  drop=True).rename_axis('company').reset_index()