如何提取csv文件里连续几行数据
时间: 2024-04-11 11:27:03 浏览: 81
要提取CSV文件中连续的几行数据,你可以使用Python中的CSV模块或Pandas库来实现。下面是两种方法的示例代码:
使用CSV模块:
```python
import csv
def extract_rows(csv_file, start_row, end_row):
rows = []
with open(csv_file, 'r') as file:
reader = csv.reader(file)
for i, row in enumerate(reader):
if i >= start_row and i <= end_row:
rows.append(row)
return rows
# 示例用法
csv_file = 'data.csv'
start_row = 2 # 起始行号
end_row = 5 # 结束行号
result = extract_rows(csv_file, start_row, end_row)
print(result)
```
使用Pandas库:
```python
import pandas as pd
def extract_rows(csv_file, start_row, end_row):
df = pd.read_csv(csv_file)
rows = df.iloc[start_row:end_row+1].values.tolist()
return rows
# 示例用法
csv_file = 'data.csv'
start_row = 2 # 起始行号
end_row = 5 # 结束行号
result = extract_rows(csv_file, start_row, end_row)
print(result)
```
以上代码示例将会从指定的CSV文件中提取出从起始行到结束行的连续数据,并将其作为列表返回。你可以根据自己的需求调整起始行和结束行的值。
阅读全文