以下是一个示例代码,用于从CSV文件中提取特定行的数据:
import csv
def extract_rows_from_csv(csv_file, rows):
extracted_rows = []
with open(csv_file, 'r') as file:
reader = csv.reader(file)
for index, row in enumerate(reader):
if index in rows:
extracted_rows.append(row)
return extracted_rows
# 示例用法
csv_file = 'data.csv'
rows_to_extract = [0, 2, 4] # 要提取的行的索引
extracted_data = extract_rows_from_csv(csv_file, rows_to_extract)
print(extracted_data)
在上面的示例中,我们定义了一个名为extract_rows_from_csv
的函数,它接受两个参数:csv_file
(CSV文件的路径)和rows
(要提取的行的索引列表)。
函数内部使用csv.reader
来打开CSV文件,并通过循环遍历每一行。如果行的索引存在于rows
列表中,就将该行添加到extracted_rows
列表中。
最后,函数返回extracted_rows
列表,其中包含从CSV文件中提取的行。
在示例用法中,我们指定了要从CSV文件中提取的行的索引,并将其作为参数传递给extract_rows_from_csv
函数。然后,将提取的数据打印出来。