Pandas在没有www.example.com _csv()的情况下写入csvdf.to

nimxete2 于 2023-01-11 发布在其他

关注(0)|答案(1)|浏览(74)

我想泛化一个写 Dataframe 函数，其中 Dataframe 和写函数可以是参数。例如：

def save_dataframe(df, write_func, filename):
    write_func(df, filename)

其用法如下：

save_dataframe(df, pd.write_csv , "test.csv")

我知道有像df.to_csv()df.to_excel()等函数，但我真正想要的是一个解决方案，以概括。
我正在寻找类似于pd.write_csv(df,filename)的东西，类似于pd.read_csv(filename)
谢啦，谢啦

pandas

来源：https://stackoverflow.com/questions/75065637/pandas-write-to-csv-without-df-to-csv

1条答案

按热度按时间

lnlaulya1#

可以在类方法中检查要保存文件扩展名

def write_func(df, filename):
  extensions_dict = {'csv':['csv'], 'excel':['xls', 'xlsx']}
  extension = filename.split(".")[-1]

  if extension in extensions_dict["csv"]:
     df.to_csv(df, filename)
  elif extension in extensions_dict["excel"]:
     df.to_excel(df, filename)
  # other extension checks ...

赞(0）回复(0）举报 2023-01-11

我来回答

Pandas在没有www.example.com _csv()的情况下写入csvdf.to

1条答案

相关问题

热门标签

最新问答