将 NumPy 数组转换为 csv 文件

在本文中，我们将看到将 NumPy 数组保存到 CSV 文件中的不同方法。 CSV 文件格式是存储数据的最简单和有用的格式

我们可以通过不同的方法将 NumPy 数组保存到 CSV 文件中

方法 1：使用Dataframe.to_csv() 。

此方法用于将 Dataframe 写入 CSV 文件。

示例：将数组转换为pandas Dataframe ，然后将其保存为 CSV 格式。

Python3

# import necessary libraries
import pandas as pd
import numpy as np
  
# create a dummy array
arr = np.arange(1,11).reshape(2,5)
  
# display the array
print(arr)
  
# convert array into dataframe
DF = pd.DataFrame(arr)
  
# save the dataframe as a csv file
DF.to_csv("data1.csv")

输出：

[[ 1  2  3  4  5]
[ 6  7  8  9 10]]

方法 2：使用numpy_array.tofile()。

此方法用于将数组写入文件。

示例：创建一个数组，然后保存到 CSV 文件中。

Python3

# import the necessary library
import numpy as np
  
# create a dummy array
arr = np.arange(1,11)
  
# display the array
print(arr)
  
# use the tofile() method 
# and use ',' as a separator
# as we have to generate a csv file
arr.tofile('data2.csv', sep = ',')

输出：

[ 1  2  3  4  5  6  7  8  9 10]

方法 3：使用numpy.savetext() 。

此方法用于将数组保存到文本文件。

示例：创建一个数组，然后另存为 CSV 文件。

Python3

# import numpy library
import numpy
  
# create an array
a = numpy.array([[1, 6, 4],
                 [2, 4, 8],
                 [3, 9, 1]])
  
# save array into csv file
numpy.savetxt("data3.csv", a, 
              delimiter = ",")

输出：