如何创建Pandas函数来读取不同大小的数据帧

Question

我正在尝试为我的工作自动化里程旅行，这涉及到读取.csv文件和使用pandas模块。问题是.csv文件的长度不同，因为每个人的行程不同。是否有任何方法可以创建一个函数，该函数可以准确读取行程数，而不考虑.csv文件的长度？csv文件在trips下面有一些额外的行，我不想读入数据帧

      a  b  c  
trip1 x  x  x  
trip2 x  x  x 
trip3 x  x  x  

      a  b  c  
trip1 x  x  x  
trip2 x  x  x 
trip3 x  x  x
trip4 x  x  x
      ...
trip9 x  x  x 

Answer 1

如果要打印顶行，请使用df.head()

import pandas as pd

def read():
     df = df.read_csv('csv_file.csv')
     df.head(10) #depends how many rows you want to print.
     print(df)

如果要从下一行打印，请使用df.tail()

import pandas as pd
def read():
    df = df.read_csv('csv_file.csv')
    df.tail(10) #depends how many rows you want to print.
    print(df)

Answer 2

我假设您要从.csv文件中读取n行，您只需执行以下操作：

pd.read_csv('path_to_file.csv', nrows=10)

这将仅读取csv中的前10行。如果要读取包含大量数据和大小的文件，这将非常有用

如果要跳过最后n行，可以执行以下操作：

pd.read_csv('path_to_file.csv', skipfooter=2)

这将始终跳过csv中的最后两行

文件：https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.read_csv.html