博客

如何用NumPy读取CSV文件

  • Post category:Python

要用NumPy读取CSV文件，可以按照以下步骤进行操作：

  1. 导入NumPy库

python
import numpy as np

  1. 使用NumPy的loadtxt()函数读取CSV文件

python
data = np.loadtxt('filename.csv', delimiter=',', skiprows=1)

解释一下这条语句：

  • 'filename.csv'是CSV文件的文件名，即要读取的文件
  • delimiter=','表示CSV文件中数据的分隔符是逗号
  • skiprows=1表示跳过CSV文件的第一行（通常是表头）

  • 读取完成后，将数据存储在data变量中

  • 可以使用data.shape来查看数据的维度

python
print(data.shape)

解释一下这条语句：

  • data是前面读取文件时存储数据的变量名

  • data.shape是一个元组，包含数据的维度信息

  • 可以使用索引操作来访问数据

python
print(data[0]) # 访问第一行数据
print(data[:, 0]) # 访问第一列数据

解释一下这条语句：

  • data[0]表示访问数据的第一行
  • data[:, 0]表示访问数据的第一列

示例一：

假设CSV文件的内容如下所示：

name,age,score
Tom,23,80
Jerry,25,85
Alice,21,90
Bob,20,95

代码如下：

import numpy as np

# 加载并读取CSV文件
data = np.loadtxt('example.csv', delimiter=',', skiprows=1)

# 查看数据的维度
print(data.shape)

# 访问第一行数据
print(data[0])

# 访问第一列数据
print(data[:, 0])

上述代码输出结果为：

(4, 3)
[23. 80.]
[23. 25. 21. 20.]

示例二：

假设CSV文件的内容如下所示：

1,2,3,4
5,6,7,8
9,10,11,12

代码如下：

import numpy as np

# 加载并读取CSV文件
data = np.loadtxt('example2.csv', delimiter=',')

# 查看数据的维度
print(data.shape)

# 访问第二行数据
print(data[1])

# 访问第三列数据
print(data[:, 2])

上述代码输出结果为：

(3, 4)
[5. 6. 7. 8.]
[ 3.  7. 11.]

综上所述，以上就是如何用NumPy读取CSV文件的完整攻略，希望可以对你有所帮助。

Tags: