将numpy数组写入具有特定列数的csv文件

Question

我有一个 numpy 形状数组 (444,445)，我需要将其转储为 csv 文件。可以通过以下方式实现：

np.savetxt('outfile.txt',array, delimiter=',',fmt="%s")

我使用fmt="%s" 选项，因为在每一行的末尾（数组的第 444 个元素是NaN）。

我想要完成的是编写一个 5 列宽的 csv 文件，总共 39,516 行（即 89 个部分，每个部分由 5 列和 444 行组成），最后是 @987654327 @ 写为第 444 行末尾的空元素。这样一来，矩阵的元素数量相等：89x5x444=444x445，即 197,580 条数据。

例如：

  1 xxxx,xxxx,xxxx,xxxx,xxxx,
  2 xxxx,xxxx,xxxx,xxxx,xxxx,
    ...
    ...
 89 xxxx,xxxx,xxxx,xxxx,
 90 xxxx,xxxx,xxxx,xxxx,xxxx,
 91 xxxx,xxxx,xxxx,xxxx,xxxx,
    ...
    ...
178 xxxx,xxxx,xxxx,xxxx,

我添加了行号以便在我的问题中更清楚。我不希望它出现在实际输出中。

什么是一种高效且 Pythonic 的方式？

目前，我正在尝试根据我的情况调整这个问题的答案：

Write to list to rows with specific number of columns.

Answer 1

希望我能理解您的要求

# Reshape it

array_.reshpe(89,444,5)

# Change it's dtype to str so you can replace NaN by white spaces

array_.astype(str)

# Replace nan by white spaces

array_[array_ == 'nan'] = ''


# Finaly, save it SEE EDIT

编辑

我认为np.savetxt 不能用于二维以上的 numpy 数组，因此，参考this answer，我们可以试试这个：

# Write the array to disk
with file('test.txt', 'w') as outfile:
    # I'm writing a header here just for the sake of readability
    # Any line starting with "#" will be ignored by numpy.loadtxt
    outfile.write('# Array shape: {0}\n'.format(array_.shape))

    # Iterating through a ndimensional array produces slices along
    # the last axis. This is equivalent to array_[i,:,:] in this case
    for data_slice in array_:

        # The formatting string indicates that I'm writing out
        # the values in left-justified columns 7 characters in width
        # with 2 decimal places.  
        np.savetxt(outfile, data_slice, fmt='%-7.2f')

        # Writing out a break to indicate different slices...
        outfile.write('# New slice\n')