Slicing Pandas DataFrame – Explained with Examples

by

Slicing is a powerful feature in Pandas that allows you to select specific parts of your DataFrame for analysis and manipulation. This blog post will explore different methods to slice a Pandas DataFrame, including selecting rows, columns, and subsets of data.

1. Slicing Rows

Slicing rows allows you to select specific rows from a DataFrame. You can use integer-based indexing or label-based indexing to achieve this.

i) Using Integer-based Indexing with iloc

The iloc method is used for integer-based indexing. It allows you to slice rows by their index positions.

Example:
Python
import pandas as pd

# Sample DataFrame
data = {
    'A': [1, 2, 3, 4, 5],
    'B': [10, 20, 30, 40, 50],
    'C': [100, 200, 300, 400, 500]
}
df = pd.DataFrame(data)

# Slicing rows using iloc
sliced_rows = df.iloc[1:4]

print(sliced_rows)

In this example, we select rows from index 1 to 3 (the end index is exclusive).

Output:
   A   B    C
1  2  20  200
2  3  30  300
3  4  40  400
ii) Using Label-based Indexing with loc

The loc method is used for label-based indexing. It allows you to slice rows by their index labels.

Example:
Python
import pandas as pd

# Sample DataFrame with custom index
data = {
    'A': [1, 2, 3, 4, 5],
    'B': [10, 20, 30, 40, 50],
    'C': [100, 200, 300, 400, 500]
}
df = pd.DataFrame(data, index=['a', 'b', 'c', 'd', 'e'])

# Slicing rows using loc
sliced_rows = df.loc['b':'d']

print(sliced_rows)

In this example, we select rows from index ‘b’ to ‘d’ (the end index is inclusive).

Output:
   A   B    C
b  2  20  200
c  3  30  300
d  4  40  400

2. Slicing Columns

Slicing columns allows you to select specific columns from a DataFrame. You can use the same methods (iloc and loc) for this purpose.

i) Using Column Names

You can slice columns by passing a list of column names.

Example:
Python
import pandas as pd

# Sample DataFrame
data = {
    'A': [1, 2, 3, 4, 5],
    'B': [10, 20, 30, 40, 50],
    'C': [100, 200, 300, 400, 500]
}
df = pd.DataFrame(data)

# Slicing columns using column names
sliced_columns = df[['A', 'C']]

print(sliced_columns)

In this example, we select columns ‘A’ and ‘C’.

Output:
   A    C
0  1  100
1  2  200
2  3  300
3  4  400
4  5  500
ii) Using Integer-based Indexing with iloc

You can also slice columns by their index positions using the iloc method.

Example:
Python
import pandas as pd

# Sample DataFrame
data = {
    'A': [1, 2, 3, 4, 5],
    'B': [10, 20, 30, 40, 50],
    'C': [100, 200, 300, 400, 500]
}
df = pd.DataFrame(data)

# Slicing columns using iloc
sliced_columns = df.iloc[:, [0, 2]]

print(sliced_columns)

In this example, we select columns at index positions 0 and 2.

Output:
   A    C
0  1  100
1  2  200
2  3  300
3  4  400
4  5  500

3. Slicing Both Rows and Columns

You can slice both rows and columns simultaneously using the iloc or loc methods.

i) Using iloc
Example:
Python
import pandas as pd

# Sample DataFrame
data = {
    'A': [1, 2, 3, 4, 5],
    'B': [10, 20, 30, 40, 50],
    'C': [100, 200, 300, 400, 500]
}
df = pd.DataFrame(data)

# Slicing both rows and columns using iloc
sliced_data = df.iloc[1:4, [0, 2]]

print(sliced_data)

In this example, we select rows from index 1 to 3 and columns at index positions 0 and 2.

Output:
   A    C
1  2  200
2  3  300
3  4  400
ii) Using loc
Example:
Python
import pandas as pd

# Sample DataFrame with custom index
data = {
    'A': [1, 2, 3, 4, 5],
    'B': [10, 20, 30, 40, 50],
    'C': [100, 200, 300, 400, 500]
}
df = pd.DataFrame(data, index=['a', 'b', 'c', 'd', 'e'])

# Slicing both rows and columns using loc
sliced_data = df.loc['b':'d', ['A', 'C']]

print(sliced_data)

In this example, we select rows from index ‘b’ to ‘d’ and columns ‘A’ and ‘C’.

Output:
   A    C
b  2  200
c  3  300
d  4  400

4. Slicing with Conditions

Slicing with conditions allows you to select data based on specific criteria. You can use boolean indexing for this purpose.

Example:
Python
import pandas as pd

# Sample DataFrame
data = {
    'A': [1, 2, 3, 4, 5],
    'B': [10, 20, 30, 40, 50],
    'C': [100, 200, 300, 400, 500]
}
df = pd.DataFrame(data)

# Slicing with conditions
sliced_data = df[df['A'] > 2]

print(sliced_data)

In this example, we select rows where the values in column ‘A’ are greater than 2.

Output:
   A   B    C
2  3  30  300
3  4  40  400
4  5  50  500

Conclusion

Slicing a Pandas DataFrame is an essential skill for data manipulation and analysis. Whether you need to select specific rows, columns, or subsets of data based on conditions, Pandas provides flexible and powerful methods to achieve this. By mastering these techniques, you can efficiently handle your data and perform complex analysis tasks with ease.

Happy slicing!

Leave a Comment