Question

我正在创建一个自定义文件编写器。我需要写出我的数组的值，逗号分隔成文件中的一行。我可以做到以下几点：

def as_csv(array):

    return ','.join([str(i) for i in array]) + '\n'

然后：

outfile.write(my_header)
outfile.write(other_stuff)
outfile.write(as_csv(array))

但我想知道这是否是最有效的方法，或者是否有更好的方法使用numpy.array_str或numpy.array_repr方法。

Answer 1

您还可以使用内置的numpy方法np.savetxt：http://docs.scipy.org/doc/numpy/reference/generated/numpy.savetxt.html

np.savetxt(outfile, array, delimiter=',')

Answer 2

我没有尝试在下面的上下文中使用np.savetxt，也许只要文件以追加模式打开就可以使用它，但这里是我尝试做的解决方案。但是，它可能不是最有效的..

def _as_csv(self, values):
    vals = ','.join([str(i) for i in values])
    return vals + '\n'

def output(self, filename, series_cnt, series_name, series_type, startdate, N, values, step = 1000):
    """ outputs data to a file """

    starttime = startdate

    num_imports = (N / step) + 1

    outfile = open(filename.format(series_cnt, i), 'w')
    outfile.write('#{0},{1},{2},\n'.format('TEST', startdate.strftime('%Y%m%d%H%M%S000'), str(num_imports)))


    for i in range(0, N, step):

        line_start = '/Test/{0},{1},{2},{3},'.format(series_name, series_type, starttime.strftime('%Y%m%d%H%M%S000'), step)
        outfile.write(line_start)
        nxt = i + step
        starttime = starttime + dt.timedelta(hours=step)

        outfile.write(self._as_csv(values[i:nxt]))

        outfile.close()

从python numpy数组中写入逗号分隔值

2 个答案: