如何將文件名插入“to_csv”

Question

我有以下 python 公式來匯總數據框中的價格。

def test(df1,keys=["price"]):
x=df1.groupby(keys).size()
x.to_csv("%s_price.txt" % (df1),sep='\t')

我嘗試如下。

test(df_Example,keys=["price"])

我想將文件名 (df_Example) 放入輸出文件中，所以我使用了 %s 但這不起作用。 有誰知道如何解決這一問題？

謝謝你。

Answer 1

您可以使用 f 字符串：

def test(df1, filename, keys=["price"]):
    x = df1.groupby(keys).size()
    x.to_csv(f"{filename}_price.txt", sep='\t')

test(df_Example, "df_Example.csv", keys=["price"])

我認為它只適用於 python 3.6+

Answer 2

我猜這個函數應該是（無法編輯）：

def test(df1,keys=["price"]):
    x=df1.groupby(keys).size()
    x.to_csv("%" % (df1),sep='\t')

一種方法是：

def test_updated(df,filename,keys=["price"]):
    x=df.groupby(keys).size()
    filename = "{}s_price.txt".format(filename)
    x.to_csv(filename,sep='\t')