如何高效地将 CSV 数据读取到 NumPy 记录数组中？

使用 NumPy 将 CSV 数据读入记录数组

在 NumPy 中，记录数组提供了一种便捷的方式来表示具有不同数据类型的结构化数据。虽然 R 提供了 read.table()、read.delim() 和 read.csv() 等函数来将 CSV 文件导入数据帧，但 NumPy 需要稍微不同的方法。

导入 CSV 有两种选项将数据写入 NumPy 中的记录数组：

1. 使用 csv.reader() 和numpy.core.records.fromrecords()：此方法涉及使用 csv.reader() 逐行读取 CSV 文件，然后使用 numpy.core.records.fromrecords 将生成的列表列表转换为记录数组().
2. 直接使用 numpy.genfromtxt() 导入：首选方法是使用numpy.genfromtxt()，支持将CSV数据导入记录数组。通过将 delimiter 关键字参数设置为逗号，您可以指定 CSV 文件以逗号分隔。

以下代码片段演示了如何使用 numpy.genfromtxt():

from numpy import genfromtxt
my_data = genfromtxt('my_file.csv', delimiter=',')

这将创建一个名为“my_data”的记录数组，其中包含 CSV 文件中的数据。

以上是如何高效地将 CSV 数据读取到 NumPy 记录数组中？的详细内容。更多信息请关注PHP中文网其他相关文章！