导入项目列表的Python脚本

Question

我有一个特定的集群代码，它以变量命名为market作为参数并执行处理。现在我想通过另一个脚本为所有市场运行此代码。假设我的集群脚本是Cmeans_omni，我从temp.py中调用它。所以，我的代码是：

###temp.py 
m= ['A','B','C', 'Other'] #market list 
for market_for_clustering in m: 
    import Cmeans_omni 

现在问题就来了那Cmeans_omni.py保存使用to_csv模块的输出。因此，代码的最后一行是：

df.to_csv(output_dir + market + output_filename,mode='w',index=False) #df is the output data generated by C_means.py 

的代码将运行所有的市场，但它的结果保存在仅去年市场（参数），这是其他的输出目录。我想保存所有市场的结果。我知道我错过了一些非常基本的东西，任何帮助将不胜感激！

Answer 1

你应该改变你的结构一点点：

Cmeans_omni.py：

def doWork(fileName): 
    ... 
    df.to_csv(fileName, mode='w', index=False) 

temp.py：

import Cmeans_omni 
markets = ['A.csv', 'B.csv', 'C.csv'] 
for m in markets: 
    Cmeans_omni.doWork(m)