python - 使用 Pandas 读取和传递 Excel 文件名

Question

我想用 Pandas 读取 excel 文件，删除标题行和第一列并将结果数据写入同名的 excel 文件中。我想对文件夹中的所有 excel 文件执行此操作。我已经编写了数据读写代码，但无法将数据保存到同名文件中。我写的代码是这样的-

import numpy as np
import pandas as pd
import os
for filename in os.listdir ('./'):
    if filename.endswith ('.xlsx'):
        df = pd.read_excel ('new.xlsx', skiprows=1)
        df.drop (df.columns [0], axis=1, inplace=True)
        df.to_csv ('new.csv', index=False)    

如何为同一文件夹中的所有 Excel 文件自动化我的代码？

Answer 1

在函数read_excel 中使用变量filename，然后通过format 创建新文件名，删除第一列可以使用DataFrame.iloc - 选择没有第一个的所有列:

for filename in os.listdir ('./'):
    if filename.endswith ('.xlsx'):
        df = pd.read_excel (filename, skiprows=1)
        df.iloc[:, 1:].to_csv('new_{}.csv'.format(filename), index=False) 

glob 的另一种解决方案，可以指定扩展名:

import glob

for filename in glob.glob('./*.xlsx'):
    df = pd.read_excel (filename, skiprows=1)
    df.iloc[:, 1:].to_csv('new_{}.csv'.format(filename), index=False)   
    #python 3.6+    
    #df.iloc[:, 1:].to_csv (f'new_{filename}.csv', index=False)