格式化 NumPy 数组

我要这个：

        SP,1,2,3
        1,1.000000e+00,2.000000e+00,3.000000e+00
        2,1.630000e+01,1.990000e+01,1.840000e+01
        3,1.630000e+01,1.990000e+01,1.840000e+01
        4,1.630000e+01,1.990000e+01,1.840000e+01

我有以下代码：

np.savetxt("Final Array.csv", my_array, fmt="%10.6e", delimiter=',')

它生产：

    1.000000e+00,2.000000e+00,3.000000e+00
    1.630000e+01,1.990000e+01,1.840000e+01
    1.630000e+01,1.990000e+01,1.840000e+01
    1.630000e+01,1.990000e+01,1.840000e+01

我已经为顶行和左列创建了数组。

如何格式化它，以便我可以保持数字格式正确，同时仍然保留最左边的列，并且顶行保持原样格式？

我使用 vstack 和 hstack 来组合它们，但“SP”也会引起问题，因为它不是浮点数。

我可以格式化它们，将其全部转换为字符串，合并然后保存txt吗？

一个简单的解决方案需要创建一个与以下长度相同的临时数组my_array和一个额外的列。

temp = np.empty((my_array.shape[0], my_array.shape[1]+1))

然后，用所需的索引填充第一列，用初始数组填充最后列：

temp[:,1:] = my_array
temp[:,0] = np.arange(1, len(my_array)+1)

为了写入标头，您必须首先以写入方式打开文件。您仍然可以将文件对象传递给np.savetxt，您只需修改格式字符串，以便第一列写为int，其他为"%10.6e":

with open('final.csv', 'w') as f:
    f.write("SP,1,2,3\n")
    np.savetxt(f, temp, fmt="%i,%10.6e,%10.6e,%10.6e",delimiter=",")

一种更具交互性的方式来定义格式字符串，具体取决于格式字符串的列数my_array is

fmt = ",".join(["%i"] + ["%10.6e"] * my_array.shape[1])
np.savetxt(f, temp, fmt=fmt, delimiter=",")