要将CSV文件的行读入NumPy数组或列表,可以使用NumPy库中的genfromtxt()
函数或Python标准库中的csv
模块。
使用NumPy的genfromtxt()
函数:
import numpy as np
data = np.genfromtxt('data.csv', delimiter=',', skip_header=1)
data.csv
是CSV文件的路径。delimiter=','
指定CSV文件中的分隔符,这里假设是逗号。skip_header=1
表示跳过CSV文件的第一行,如果CSV文件没有标题行,可以省略这个参数。使用Python标准库中的csv
模块:
import csv
data = []
with open('data.csv', 'r') as file:
reader = csv.reader(file)
next(reader) # 跳过第一行,如果CSV文件没有标题行,可以省略这行
for row in reader:
data.append(row)
data.csv
是CSV文件的路径。csv.reader()
函数用于创建一个CSV文件的阅读器对象。next(reader)
用于跳过第一行,如果CSV文件没有标题行,可以省略这行。data
列表中。以上两种方法都可以将CSV文件的行读入NumPy数组或列表中,具体选择哪种方法取决于个人偏好和需求。
领取专属 10元无门槛券
手把手带您无忧上云