python - 将 NumPy 数组/向量保存到 CSV

Question

我有一些数字(大约 550,000)，我尝试将它们保存在 CSV 文件中。我的值只能是1或2。但它保存为:

['2.000000000000000000e+00']
['1.000000000000000000e+00']
['2.000000000000000000e+00']
['2.000000000000000000e+00']
...

我的代码是:

import numpy as np

def save_in_scv_format(My_Labels):
    K = []
    for i in range(len(My_Labels)):
        K.append(My_Labels[i])

    np.savetxt('My_labels.csv', K, delimiter = ',')

My_labels 是一个具有整数值 1 或 2、长度为 550,000 的向量。

如何将这些值保存为 1 或 2？

Answer 1

您可以更改输出中数值的格式。来自manual :

fmt : str or sequence of strs, optional

   A single format (%10.5f), a sequence of formats, or a multi-format string,
   e.g. ‘Iteration %d – %10.5f’, in which case delimiter is ignored.

所以尝试一下:

np.savetxt('My_labels.csv', K, delimiter = ',', fmt='%d')

但是，此代码还存在其他问题。

import numpy as np

def save_in_csv_format(My_Labels):
    np.savetxt('My_labels.csv', My_Labels, delimiter = ',', fmt='%d')

这应该做完全相同的事情，而且效率更高。