pandas get position of a given index in DataFrame

by

For your first question:

base = df.index.get_indexer_for((df[df.A == 2].index))

or alternatively

base = df.index.get_loc(18)

To get the surrounding ones: 

mask = pd.Index(base).union(pd.Index(base - 1)).union(pd.Index(base + 1))

I used Indexes and unions to remove duplicates. You may want to keep them, in which case you can use np.concatenate

Be careful with matches on the very first or last rows 🙂

More Related Contents:

  1. How can I pivot a dataframe?
  2. GroupBy pandas DataFrame and select most common value
  3. String concatenation of two pandas columns
  4. Reshape wide to long in pandas
  5. How to create a large pandas dataframe from an sql query without running out of memory?
  6. Find column name in pandas that matches an array
  7. How do convert a pandas dataframe to XML?
  8. How to add hovering annotations to a plot
  9. Sort a pandas dataframe series by month name
  10. Count frequency of values in pandas DataFrame column
  11. Strip / trim all strings of a dataframe
  12. How to iterate over consecutive chunks of Pandas dataframe efficiently
  13. How to flatten multilevel/nested JSON?
  14. Neural Network LSTM input shape from dataframe
  15. Joining pandas DataFrames by Column names
  16. pandas get the row-wise minimum value of two or more columns
  17. Diff of two Dataframes
  18. Apply function to pandas DataFrame that can return multiple rows
  19. Write to StringIO object using Pandas Excelwriter?
  20. Scikit Learn OneHotEncoder fit and transform Error: ValueError: X has different shape than during fitting
  21. “Anti-merge” in pandas (Python)
  22. Pandas – write Multiindex rows with to_csv
  23. Writing to a file on SFTP server opened using Paramiko/pysftp “open” method is slow
  24. Convert Pandas series containing string to boolean
  25. How to annotate a seaborn barplot with the aggregated value
  26. Pandas groupby and aggregation output should include all the original columns (including the ones not aggregated on)
  27. Merging two tables with millions of rows in Python
  28. Selecting a range of columns in a dataframe
  29. Pandas data frame to dictionary of lists
  30. Pandas min() of selected row and columns

Leave a Comment