如何将结果写入输出文件

Question

我在尝试将简单结果输出到输出文件时遇到问题。在这里，我尝试使用 numpy 制作一些虚假数据，但这会导致其自身出现问题，我不断收到各种错误，基本上与索引不可订阅或标量变量索引无效有关。

我想要的是一个输出 csv 文件，内容如下：

LineID  ColA    ColB    ColC
1  8  2 93
2  7  3 86
3  7  3 89
4  9  2 90

使用的代码：

import csv
import numpy as np



z=np.array([[1,8,2,93],[2,7,3,86],[3,7,3,89],[4,9,2,90]])

#b=np.array(z)
b=z
print('\n','b is : ','\n', b, '\n')

nummy=z[:,0]
ColA_results = z[:,1]
ColB_results = z[:,2]
ColC_results = z[:,3]
print('nummy: ',nummy)
print('ColA_results: ',ColA_results)
print('ColB_results: ',ColB_results)
print('ColC_results: ',ColC_results)

results_filename = (r'/Users/imagexpertinc/Desktop/steve_results/demo_03.csv')
with open(results_filename, 'w', newline='') as filey: ## Python 3... with open('/pythonwork/thefile_subset11.csv', 'w', newline='') as outfile:
    csv_writer = csv.writer(filey, delimiter = '\t', lineterminator='\n')
    csv_writer.writerow(['LineID', 'ColA','ColB','ColC'])
filey.close()




size_of_z=len(z)

indy = range(size_of_z)
print('indy is: ',indy)


for i in indy:
    print('i is :',i)
    print('nummy[i] is: ',nummy[i])
    #nummy = nummy[i]
    ColA_results = ColA_results[i]
    ColB_results = ColB_results[i]
    ColC_results = ColC_results[i]

    with open(results_filename, 'a') as filey: ## Python 3... with open('/pythonwork/thefile_subset11.csv', 'w', newline='') as outfile:
        csv_writer = csv.writer(filey, delimiter = '\t', lineterminator='\n')
        csv_writer.writerow([nummy, ColA_results, ColB_results, ColC_results])

关于我在这里做错了什么的任何指示？

Answer 1

你想多了。 Python 将 numpy 数组视为可迭代对象的可迭代对象，并且可以直接传递给 csv.writer.writerows（注意行 s 中的 s） :

import csv
import numpy as np



z=np.array([[1,8,2,93],[2,7,3,86],[3,7,3,89],[4,9,2,90]])

results_filename = ('/Users/imagexpertinc/Desktop/steve_results/demo_03.csv')
with open(results_filename, 'w', newline='') as filey:
    csv_writer = csv.writer(filey, delimiter = '\t', lineterminator='\n')
    csv_writer.writerow(['LineID', 'ColA','ColB','ColC'])

    csv_writer.writerows(z)

Answer 2

你可以试试：

import csv
import numpy as np

z=np.array([[1,8,2,93],[2,7,3,86],[3,7,3,89],[4,9,2,90]])

nummy=z[:,0]
ColA_results = z[:,1]
ColB_results = z[:,2]
ColC_results = z[:,3]

results_filename = (r'/Users/imagexpertinc/Desktop/steve_results/demo_03.csv')
with open(results_filename, 'w', newline='') as filey: ## Python 3... with open('/pythonwork/thefile_subset11.csv', 'w', newline='') as outfile:
    csv_writer = csv.writer(filey, delimiter = '\t', lineterminator='\n')
    csv_writer.writerow(['LineID', 'ColA','ColB','ColC'])

    for i in range(z.shape[0]):
        csv_writer.writerow([nummy[i], ColA_results[i], ColB_results[i], ColC_results[i]])
filey.close()

如何将结果写入输出文件

How to write results to an output file

python

csv

file-io