python-3.x - 使用列 Pandas 的特定单词提取行直到特定行

Question

我有一个这样的数据框，

   Name    Product    Quantity
0  NaN     1010       10
1  NaN     2010       12
2  NaN     4145       18
3  NaN     5225       14
4  Total   6223       16
5  RRA     7222       18
6  MLQ     5648       45

现在，我需要提取包含行的行/新数据帧，直到 Name 列中的 Total。

需要的输出:

       Name    Product    Quantity
    0  NaN     1010       10
    1  NaN     2010       12
    2  NaN     4145       18
    3  NaN     5225       14

我试过了，

df[df.Name.str.contains("Total", na=False)]

这暂时没有帮助。任何建议都会很棒。

Answer 1

选择True值所在索引，使用df.iloc切片:

df_new=df.iloc[:df.loc[df.Name.str.contains('Total',na=False)].index[0]]

或使用 series.idxmax()这允许您获得最大值的索引(True/False 的最大值为 True):

df_new=df.iloc[:df.Name.str.contains('Total',na=False).idxmax()]

print(df_new)

  Name  Product  Quantity
0  NaN     1010        10
1  NaN     2010        12
2  NaN     4145        18
3  NaN     5225        14