subset dataframe based on index value in python

To subset a dataframe based on index value, you can use the .loc accessor.

Here's an example:


main.py
import pandas as pd

# Create a sample dataframe
df = pd.DataFrame({
    'name': ['Alice', 'Bob', 'Charlie', 'Dave'],
    'age': [25, 30, 35, 40],
    'salary': [50000, 60000, 70000, 80000]
})

# Set the 'name' column as the index
df.set_index('name', inplace=True)

# Subset the dataframe based on index value
subset_df = df.loc['Bob':'Dave']

# Print the subset dataframe
print(subset_df)
391 chars
18 lines

In the code above, we first create a sample dataframe with a 'name', 'age', and 'salary' column. We then set the 'name' column as the index. To subset the dataframe based on index value, we use the .loc accessor and pass in the range of index values we are interested in (in this case, 'Bob' to 'Dave'). Finally, we print the subset dataframe.

The output will be:


main.py
         age  salary
name               
Bob       30   60000
Charlie   35   70000
Dave      40   80000
104 chars
6 lines

similar python code snippets

for loop on a dataframe variables in python

loop on a dataframe rows in python

append a line at the end of a dataframe in python

add a dataframe serie to a dictionnary in python

create a matrix in pandas in python

covnert all dataframe values from int to str in python

merge two series in a dataframe with special function for each column in python

merge two series in a dataframe in python

iterate dataframe fast way in python

initialize dataframe in python

related categories

pandas

dataframe