python - 在 Pandas Dataframes 中查找单个文件的第一行和最后一行

Question

我有一个由多个 .fits 文件组成的 Pandas Dataframe，每个文件都包含多个带有单独标签的列。我想提取一列并创建包含所述列的第一行和最后一行的变量，但我很难为单个 .fits 文件而不仅仅是整个 Dataframe 完成此操作。任何帮助，将不胜感激! :)

这是我读取文件的方式:

path = '/Users/myname/folder/'
m = [os.path.join(dirpath, f)
    for dirpath, dirnames, files in os.walk(path)
    for f in fnmatch.filter(files, '*.fits')]

^^^ 这递归地搜索我的目录，其中包含许多子文件夹中的多个 .fits 文件。

dataframes = []
for ii in range(0,len(m)):
    data = pd.read_csv(m[ii], header = 'infer', delimiter = '\t')
    d = pd.DataFrame(data)
    top = d['desired_column'].head()
    bottom = d['desired_column'].tail()
    First_and_Last = pd.concat([top,bottom])

我尝试对 Pandas Dataframes 使用 .head 和 .tail 命令，但我不确定如何正确使用它来实现我想要的。对于我在 fits 文件中的读取方式，以下代码给出了最开始的几行和最后几行(确切地说是 5 行，头部和尾部的默认值为 5)，如下所示:

0       2.456849e+06
1       2.456849e+06
2       2.456849e+06
3       2.456849e+06
4       2.456849e+06
1118    2.456852e+06
1119    2.456852e+06
1120    2.456852e+06
1121    2.456852e+06
1122    2.456852e+06

我想要做的是尝试为我想要的特定列获取每个 .fits 文件的第一行和最后一行，而不仅仅是包含 .fits 文件的 Dataframe。按照我阅读 .fits 文件的方式，Dataframe 似乎将所有文件连接在一起。关于如何实现此目标的任何提示？

Answer 1

如果你只想要第一行:

top = d['desired_column'].head(1)

如果你只想要最后一行:

bottom = d['desired_column'].tail(1)

我没有发现“Dataframe 似乎有点将所有文件连接在一起”的问题。你能澄清一下这个问题吗？
顺便说一句，在 data = pd.read_csv(m[ii], header = 'infer', delimiter = '\t') 之后，data 已经是一个 DataFrame 。因此，d = pd.DataFrame(data) 是不必要的。