Pandas 在沒有 df.to_csv() 的情況下寫入 csv

Question

我想概括一個 write dataframe function 其中 dataframe 和 write 函數可以是參數。 例如：

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)

謝謝

Answer 1

您可以檢查文件擴展名以保存在該 class 方法中

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 ...