python - 根据数据框列中的某个值获取所有数据框

Question

我DataFrame 看起来像这样：

       import numpy as np
       import pandas as pd

       df=pd.DataFrame([['d',5,6],['a',6,6],['index',5,8],['b',3,1],['b',5,6],['index',6,7],
            ['e',2,3],['c',5,6],['index',5,8]],columns=['A','B','C'])

我想选择之间的所有行index并创建许多数据框

我想获得所有：

数据框1：

         A  B   C
 1       a  6   6
 2   index  5   8
 3      3   b   3   

数据框 2

                A   B   C
         4      b   5   6
         5  index   6   7
         6      c   2   3

数据框3：

                   A   B    C
              7    c    5   6
              8 index   5   8
              9    4    3   1

数据框4：

                            A  B   C

                     11    5    2   3
                     12 index   4   2
                     13   1     2   5

标签： pythonpandasnumpydataframe

Answer 1

index_list = df.index[df['A'] == 'index'].tolist() # create a list of the index where df['A']=='index'
new_df = [] # empty list for dataframes
for i in index_list: # for loop
    try:
        new_df.append(df.iloc[i-1:i+2])
    except:
        pass

这将创建一个数据框列表，您可以调用它们new_df[0] new_df[1]或使用循环将它们打印出来：

for i in range(len(new_df)):
    print(f'{new_df[i]}\n')

       A  B  C
1      a  6  6
2  index  5  8
3      b  3  1

       A  B  C
4      b  5  6
5  index  6  7
6      e  2  3

       A  B  C
7      c  5  6
8  index  5  8