使用Pandas读取CSV文件指定的前几行可以通过read_csv()
方法的nrows
参数来指定。具体的攻略如下:
- 导入Pandas库
import pandas as pd
- 使用
read_csv()
方法读取CSV文件,并指定nrows参数
df = pd.read_csv('file.csv', nrows=5)
其中,'file.csv'
表示CSV文件的文件名或文件路径,nrows=5
表示指定读取前5行数据。这里将读取到的数据赋值给变量df
。
- 输出读取到的前5行数据
print(df)
在上述代码运行后,将会输出CSV文件的前5行数据。
示例1:读取GitHub上Python的star数量前5的项目
import pandas as pd
import requests
# 爬取GitHub上Python的star数量前1000的项目
url = 'https://api.github.com/search/repositories?q=language:python&sort=stars&order=desc&per_page=1000'
response = requests.get(url)
data = response.json()['items']
# 将爬取到的数据保存至CSV文件
df = pd.DataFrame(data)
df.to_csv('github_python_top1000.csv', index=False)
# 读取CSV文件的前5行数据
df = pd.read_csv('github_python_top1000.csv', nrows=5)
# 输出读取到的前5行数据
print(df)
示例2:读取本地CSV文件的前3行数据
import pandas as pd
# 创建本地CSV文件
with open('local_file.csv', 'w') as f:
f.write('name,age,gender\n')
f.write('Alice,22,Female\n')
f.write('Bob,21,Male\n')
f.write('Charlie,23,Male\n')
# 读取CSV文件的前3行数据
df = pd.read_csv('local_file.csv', nrows=3)
# 输出读取到的前3行数据
print(df)
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:使用实现pandas读取csv文件指定的前几行 - Python技术站