python - 如何在 Python 中将几列复制到另一个文件？

Question

我在这里找不到类似的问题或现有的解决方案。

我有一个用 Python 编写的代码，它向我展示了 file1 中的哪些列应该被拒绝。我想编写一个简短的脚本，将有效列+最后一列打印到文件2。

File1 (input):
col1,col2,col3,col4,col5,...,col(n-1),coln

Python code chooses some of them, for example col3, col5, col6, col20

File2 (output):
col3,col5,col6,col20,coln

我正在考虑使用 awk 创建一个循环，但我不确定如何在每次迭代时仅将新列添加到一个文件中：

cmd1='''awk -F "," '{print $indices[f]}' data.csv > temp[i].csv'''
cmd2='''awk -F "," {'{print $0}' temp[i].csv '{print $[i+1]}'} data.csv} > temp[i+1].csv'''

print("Feature ranking:")

for f in range(X.shape[1]):
    if(importances[indices[f]]>0.01):
         print("%d. feature %d (%f)" % (f + 1, indices[f], importances[indices[f]]))
    while(importances[indices[f]]>0.01):
        os.system(cmd1)
        os.system(cmd2)

标签： pythonloopsawk