从Array写入CSV文件

Question

我正在尝试使用writer.writerow将数据从数组提供给csv文件。我有一个数组sum_balance，显然我需要将它转换为numpy数组才能使用writer.writerow函数。继承我的代码：

numpy_arr = array(sum_balance)
with open("output.csv", "wb") as csv_file:
    writer = csv.writer(csv_file, delimiter=',')
    for element in numpy_arr:
        writer.writerow(element)
csv_file.close()

但我仍然收到错误：writer.writerow(element)_csv.Error: iterable expected, not numpy.float64

Answer 1

numpy迭代器似乎是迭代元素而不是行，这就是你得到错误的原因。但是，有一种更简单的方法可以实现您尝试做的事情：numpy有一个例程savetxt，可以将ndarray写入csv文件：

output_array = np.array(my_data)
np.savetxt("my_output_file.csv", output_array, delimiter=",")