使用动态 CSV 命名将循环导出到 CSV

Question

我对 Python 比较陌生（我的大部分经验都是在 SAS 方面），所以请多多包涵。

我正在尝试从现有数据集创建多个 CSV，并根据定义的列表导出它们。 CSV 的命名应根据相应的列表值动态命名。

我已经尝试了很多东西 - 主要是在黑暗中刺伤 - 没有任何效果。见下面的代码

cc = ['AD-1','AD-2','AD-3'] #the list I want it to cycle through
for index in range(len(cc)): 
    df1_cc = df[df['charge'].isin(cc)] #df is predefined 
    #set "charge" as the index variable so you can aggregate on it           
    df1_cc = df1_cc.set_index('charge')
    df1_cc
    #sum up values based on individual values of 'charge'
    table1_cc = df1_cc.sum(level='charge')
    table1_cc
    #output to CSV
    table1_cc.to_csv(r"C:\Users\etc\table1_"+cc+".csv")

注意，cc（AD-1、AD-2 和 AD-3）中的值包含在 'charge' 等中

我得到的唯一错误是：

table1_cc.to_csv(r"C:\Users\etc\"+cc+".csv")

我得到的错误是： TypeError：只能将str（不是“list”）连接到str

输出应该是 3 个文件：table1_AD-1.csv、table1_AD-2.csv 和 table1_AD-3.csv，每个文件都应该包含每个单独的总和值（同样，该部分有效。真正的问题是循环并将 cc 中每个单独值的输出导出到 CSV）。

感谢任何帮助！

Answer 1

to_csv的最后一行需要改一下

cc = ['AD-1','AD-2','AD-3'] #the list I want it to cycle through
for index in range(len(cc)): 
    df1_cc = df[df['charge'].isin([cc[index]])] #df is predefined 
    #set "charge" as the index variable so you can aggregate on it           
    df1_cc = df1_cc.set_index('charge')
    df1_cc
    #sum up values based on individual values of 'charge'
    table1_cc = df1_cc.sum(level='charge')
    table1_cc
    #output to CSV
    table1_cc.to_csv(r"C:\Users\etc\table1_"+cc[index]+".csv")

Answer 2

您还可以像这样遍历您的 cc 列表：

cc_list = ['AD-1','AD-2','AD-3'] #the list I want it to cycle through

for index,cc in enumerate(cc_list): 
    df1_cc = df[df['charge'].isin([cc])] #df is predefined 
    #set "charge" as the index variable so you can aggregate on it           
    df1_cc = df1_cc.set_index('charge')
    df1_cc
    #sum up values based on individual values of 'charge'
    table1_cc = df1_cc.sum(level='charge')
    table1_cc
    #output to CSV
    table1_cc.to_csv(r"C:\Users\etc\table1_{}.csv".format(cc))