python - 从特定索引中重新选择 Pandas 数据框

Question

我有一个数据框，我想使用索引“重新选择”，因为缺乏更好的术语。

Column1 | Column2 | Column3
   1    |    2    |    3
   4    |    5    |    6
Data    |         |  
Header  | Header2 | Header3
   7    |    8    |    9

我想选择此数据帧，但特别是在索引 4,0 处，以便它看起来像:

Header  | Header2 | Header3
   7    |    8    |    9

我知道我可以做类似的事情

df.iloc[[4,5], [0,1,2]]

但问题是我并不总是知道这个索引在每个数据帧中的位置，我唯一拥有的信息是我想要开始的值的索引。

Answer 1

对于 future 的旅行者，我最终以一种有点奇怪的方式做到了这一点。首先我得到了我感兴趣的专栏:

columns = [x for x in df.iloc[4].values.flatten().tolist() if str(x) != 'nan']

然后我对我需要的数据进行了切片:

data = df.iloc[5:, 0:2].values

最后创建了一个新的数据框:

df = pd.DataFrame(data, columns=columns)

在我设置列的地方，4是列所在行的索引(我已经知道的数字)。当我设置数据时，5:表示第5行及以后的所有数据(因为我不知道它在哪里结束)，0:2是第1列- 3.

我不确定这是否是最好的方法，但它适合我需要它做的事情。