python - 将 float 保存到文件

Question

我计算了几个列表的平均值。现在我想将数据保存到 txt 或 csv 文件中。这是我的方法:

k=range(8,15)
for i in k:
    q=range(i)
    g=mean(q)    
    print g        
    savetxt('mean_values.txt', g)

但这给了我 IndexError: tuple index out of range。我认为这是因为 savetxt 需要一个数组，但是 g conatins float 。但即使我定义了一个数组g_，我最终还是得到了同样的错误:

k=range(8,15)
for i in k:
    q=range(i)
    g=mean(q)    
    g_=array(g)    
    print g_
    savetxt('mean_values.txt', g_)

诀窍在哪里？

Answer 1

在您的第一次尝试中，您试图保存一个 numpy.float64 变量，文档说 numpy.savetxt需要一个 array_like 对象。

在您的第二次尝试中，您错过了用于指定矩阵 g_=array([g]) 的方括号，但是如果您将 txt 保存在循环中，您每次都会覆盖输出文件。

我想这就是你想要的:

import numpy as np

g = list()
k = range(8,15)
for i in k:
    q = range(i)
    g.append(np.mean(q))

np.savetxt('myfile.txt', np.array(g), fmt='%.2f')

myfile.txt 的输出:

3.50
4.00
4.50
5.00
5.50
6.00
6.50