欢迎访问宙启技术站
智能推送

使用fromfile()函数读取CSV文件数据的示例代码

发布时间:2023-12-17 19:19:37

fromfile()函数是NumPy库中的一个函数,用于从文件中读取数据并创建一个数组。在读取CSV文件时,可以使用该函数来读取数据,并将数据存储在NumPy数组中。

下面是使用fromfile()函数读取CSV文件数据的示例代码:

import numpy as np

# 使用fromfile()函数读取CSV文件数据
data = np.fromfile('data.csv', dtype=float, sep=',')

# 将一维数组转化为二维数组
data = data.reshape((len(data)//3, 3))

# 打印数据
print(data)

在这个例子中,我们假设有一个名为"data.csv"的CSV文件,它包含3列数据。我们将使用fromfile()函数读取该文件的数据,并将数据存储在名为"data"的NumPy数组中。

在读取数据后,我们可以使用reshape()函数将一维数组转化为二维数组。在这个例子中,我们将数组的长度除以3,以将数据分成三列。

最后,我们打印数据以确认数据是否正确读取。

下面是一个使用示例:

假设有一个名为"data.csv"的CSV文件,内容如下:

1,2,3
4,5,6
7,8,9

使用上述代码读取该文件后,输出结果如下:

array([[1., 2., 3.],
       [4., 5., 6.],
       [7., 8., 9.]])

这里我们得到了一个3行3列的二维数组,每一行代表一条数据记录,每一列代表一种数据类型。

这个示例展示了如何使用fromfile()函数读取CSV文件数据并将数据存储在NumPy数组中。可以根据实际情况进行相应的修改,以适应不同的数据需求。