如何将结果写入输出文件

时间:2020-01-10 11:01:04

标签: python csv file-io

我在尝试将简单结果输出到输出文件时遇到问题。在这里,我尝试使用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])

这里有关于我在做什么错的指针吗?

2 个答案:

答案 0 :(得分:1)

您已经思考了这个问题。 numpy数组在Python中被视为可迭代的可迭代对象,可以直接传递给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) ):

{{1}}

答案 1 :(得分:1)

您可以尝试:

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()