如何解决列表理解以打开 CSV 文件 - Python
我正在尝试创建一个列表理解来对两个 .csv 文件执行读/写操作。我需要以 .xlsx 格式打开原始文件,然后将它们写入 .csv 并将它们声明为用于基本读/写功能的 Pandas 数据帧。我的主要问题是使用“打开”函数来打开相应的 .csvs,而无需对文件名本身的后缀进行硬编码。下面是我目前用来将 .xlsx 文件写入 .csv 格式的代码,它是可行的,但如果可能的话,我想将其转换为列表理解……非常感谢任何见解!
原始脚本(“for”循环):
import os
import glob
import pandas as pd
path = "C:\\users\\mdl518\\Desktop\\"
for file in glob.glob(os.path.join(path,"*.xlsx")): # loop through the .xlsx files
if file.endswith("firstname.xlsx"):
xlsx_1 = pd.read_excel(file)
with open(os.path.join(path,file[:-5] + '_test1.csv'),'w',newline='') as f:
xlsx_1.to_csv(f,index=None,header=True)
for file in glob.glob(os.path.join(path,"*.csv")):
df=pd.read_csv(file)
if file.endswith("secondname.xlsx"):
xlsx_2 = pd.read_excel(file)
with open(os.path.join(path,file[:-5] + '_test2.csv'),newline='') as f:
xlsx_2.to_csv(f,"*.csv")):
if file.endswith("test2.csv"):
df2=pd.read_csv(file)
所需的脚本(更清晰的列表理解,类似于以下内容):
output = [pd.read_excel(i),pd.to_csv for i in glob.glob(os.path.join(path,"*.xlsx"))]
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。