Numpy 将多个数组保存到带列名的CSV文件中

阅读更多：Numpy 教程

背景

Numpy 是 Python 中一款广泛使用的数值计算库，其提供了丰富的数据处理和科学计算功能。在数据处理过程中，我们可能需要将多个 Numpy 数组保存到一个 CSV 文件中，并在头部加上列名。本文将演示如何使用 Numpy 实现将多个数组保存到 CSV 文件中。

实现方法

在 Numpy 中，我们可以使用 np.column_stack 将多个数组沿着列合成一个新的数组。例如：

import numpy as np

a = np.array([1,2,3])
b = np.array([4,5,6])
c = np.column_stack((a, b))

print(c)

输出结果为：

[[1 4]
 [2 5]
 [3 6]]

接着，我们可以使用 Numpy 提供的 np.savetxt 将数组保存到 CSV 文件。该函数的用法如下：

np.savetxt(fname, X, fmt='%.18e', delimiter=',', newline='\n', header='', footer='', comments='# ')

其中：

fname：要保存的文件名；
X：要保存的数组；
fmt：格式化字符串；
delimiter：分隔符；
newline：行结束符；
header：头部信息；
footer：尾部信息；
comments：注释标记。

因此，我们可以使用如下代码将两个数组保存到 CSV 文件中：

import numpy as np

a = np.array([1,2,3])
b = np.array([4,5,6])
c = np.column_stack((a, b))

header = "index,a,b"
np.savetxt('data.csv', c, delimiter=',', header=header, comments='')

with open('data.csv') as f:
    data = f.read()

print(data)

其中，header 为头部信息，comments 为空字符串，表示不加注释标记。输出结果为：

# index,a,b
1.000000000000000000e+00,4.000000000000000000e+00,5.000000000000000000e+00
2.000000000000000000e+00,5.000000000000000000e+00,6.000000000000000000e+00
3.000000000000000000e+00,6.000000000000000000e+00,7.000000000000000000e+00

可以看出，数据成功保存到了 file.csv 文件中，并在头部加上了列名。