将 NumPy 数组转换为 csv 文件


Numpy 是 Python 编程语言中的一个库,缩写为 Numerical Python。它用于在较短的时间内进行数学、科学和统计计算。numpy 函数的输出将是一个数组。使用名为 savetxt() 的函数,可以将 numpy 函数创建的数组存储在 CSV 文件中。

将 NumPy 数组转换为 .csv 文件

CSV 是逗号分隔值 (Comma Separated Values) 的缩写。这是数据科学中最广泛使用的文件格式。它以表格格式存储数据,其中列保存数据字段,行保存数据。

CSV 文件中的每一行都用逗号或分隔符字符分隔,用户可以自定义分隔符。在数据科学中,我们需要使用 pandas 库来处理 csv 文件。

还可以使用 numpy 库的 loadtxt() 函数将 csv 文件中的数据恢复为 numpy 数组格式。

语法

以下是使用 **savetxt()** 函数将 numpy 数组转换为 csv 文件的语法。

numpy.savetxt(file_name,array,delimiter = ‘,’)

其中,

  • **numpy** 是库的名称。

  • **savetxt** 是用于将数组转换为 .csv 文件的函数。

  • **file_name** 是 csv 文件的名称。

  • **array** 是需要转换为 .csv 文件的输入数组。

示例

为了将数组元素转换为 csv 文件,我们必须将文件名和输入数组作为输入参数传递给 **savetxt()** 函数,并附带 delimiter = ‘,’ ,然后数组将转换为 **csv** 文件。以下是一个示例。

import numpy as np
a = np.array([23,34,65,78,45,90])
print("The created array:",a)
csv_data = np.savetxt("array.csv",a,delimiter = ",")
print("array converted into csv file")
data_csv = np.loadtxt("array.csv",delimiter = ",")
print("The data from the csv file:",data_csv)

输出

The created array: [23 34 65 78 45 90]
array converted into csv file
The data from the csv file: [23. 34. 65. 78. 45. 90.]

示例

让我们再看一个示例,其中我们将二维数组数据转换为 CSV 文件,使用 numpy 库的 savetxt() 函数。

import numpy as np
a = np.array([[23,34,65],[78,45,90]])
print("The created array:",a)
csv_data = np.savetxt("2d_array.csv",a,delimiter = ",")
print("array converted into csv file")
data_csv = np.loadtxt("array.csv",delimiter = ",")
print("The data from the csv file:",data_csv)

输出

The created array: [[23 34 65]
[78 45 90]]
array converted into csv file
The data from the csv file: [[23. 34. 65.]
[78. 45. 90.]]

示例

在下面的示例中,我们使用“_”作为 CSV 文件的分隔符,而不是“,”。

import numpy as np
a = np.array([[23,34,65,87,3,4],[78,45,90,53,5,3]],int)
print("The created array:",a)
csv_data = np.savetxt("2d_array.csv",a,delimiter = "_")
print("array converted into csv file")
data_csv = np.loadtxt("2d_array.csv",delimiter = "_")
print("The data from the csv file:",data_csv)

输出

以下是数组转换为 csv 文件后的输出。

The created array: [[23 34 65 87 3 4]
[78 45 90 53 5 3]]
array converted into csv file
The data from the csv file: [[23. 34. 65.]
[78. 45. 90.]]

更新于: 2023年8月9日

648 次浏览

开启你的 职业生涯

通过完成课程获得认证

开始学习
广告