Numpy Python 将Numpy数组写入CSV文件

在本文中，我们将介绍如何使用Numpy python将一个Numpy数组写入到CSV文件中。

阅读更多：Numpy 教程

什么是Numpy?

Numpy是一个在Python中用来优化数组操作的库。它支持各种数组和矩阵计算，包括数学、逻辑、形状操作、排序、选择、输入输出、离散傅里叶变换、基本线性代数、基本统计操作和随机模拟等等。它是Python中进行科学计算的核心库之一。

什么是CSV?

CSV(Comma Separated Values) 是一种纯文本格式，用于将表格数据存储为文本文件。每行数据都是一条记录，由逗号分隔的字段组成。CSV文件可以使用任何文本编辑器打开，并且可以轻松转换为其他格式。由于CSV文件是文本文件，因此它们易于处理和转换，是数据共享和导入导出的一种常见格式。

将Numpy数组写入CSV文件

写入CSV文件需要使用Python中的csv库。我们可以使用此库的writer对象来将数据写入CSV文件。下面是一个简单的例子，展示如何使用Numpy和csv库将数据写入CSV文件。在这个例子中，我们创建一个（3，4）的Numpy数组，并将其写入到CSV文件中。

import numpy as np
import csv

# Create an array
data = np.array([[10, 20, 30, 40], [50, 60, 70, 80], [90, 100, 110, 120]])

# Write the array to CSV
with open('data.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerows(data)

在上述示例中，我们使用numpy.array() 和 csv库的writer对象将数据写入CSV文件。现在，我们打开CSV文件，并查看其内容：

10,20,30,40
50,60,70,80
90,100,110,120

可以看到，这个CSV文件包含3行和4列。每一行都是由逗号分隔的数据值组成的，代表给定行的数据。

将Numpy数组转换为CSV格式的字符串

除了将Numpy数组写入到CSV文件中，我们还可以将其转换为CSV格式的字符串。下面是一个例子，展示如何将Numpy数组转换为字符串。

import numpy as np

# Create an array
data = np.array([[10, 20, 30, 40], [50, 60, 70, 80], [90, 100, 110, 120]])

# Convert the array to a CSV string
csv_string = np.array2string(data, separator=',', formatter={'float_kind':lambda x: "%.2f" % x})

print(csv_string)

在上述示例中，我们使用numpy中的array2string()函数将Numpy数组转换为CSV格式的字符串。此函数接受一个分隔符参数(separator)用于标识不同的字段，并指定了一个格式化器(formatter)来对float类型的数据进行格式化。最后，我们打印生成的CSV字符串：

[[10, 20, 30, 40],
 [50, 60, 70, 80],
 [90,100,110,120]]

可以看到，生成的字符串与我们前面创建的数据数组相同。

在写入CSV文件时指定格式

有时，我们可能想更改CSV文件中数据的格式。例如，在数据位数较大时，可以使用科学计数法表示。在使用csv库写入CSV文件时，我们也可以指定相应的数据格式。下面是一个示例：

import numpy as np
import csv

# Create an array
data = np.array([[15000, 20000], [30000, 40000]])

# Write the array to CSV
with open('data.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile, delimiter=',', quotechar='"', quoting=csv.QUOTE_MINIMAL, doublequote=False)
    writer.writerows(data)