sort_index() for index sorting in Pandas - Time & Space Complexity

Time Complexity: sort_index() for index sorting

O(n log n)

Understanding Time Complexity

When we sort data by its index using pandas, it takes some time depending on how much data there is.

We want to understand how the time needed grows as the data size grows.

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import pandas as pd

df = pd.DataFrame({
    'value': [10, 20, 30, 40]
}, index=[3, 1, 4, 2])

sorted_df = df.sort_index()

This code creates a small table with an unsorted index and then sorts the table by that index.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: The sorting algorithm compares and rearranges index values.
• How many times: It processes each index value multiple times depending on the sorting method.

How Execution Grows With Input

As the number of rows grows, the sorting work grows faster than just the number of rows.

Input Size (n)	Approx. Operations
10	About 30 to 40 operations
100	About 700 to 800 operations
1000	About 10,000 to 12,000 operations

Pattern observation: The operations grow faster than the input size, roughly multiplying by a bit more than the input size each time.

Final Time Complexity

Time Complexity: O(n log n)

This means sorting takes more time as data grows, but not as fast as checking every pair; it grows in a balanced way.

Common Mistake

[X] Wrong: "Sorting by index takes the same time no matter how many rows there are."

[OK] Correct: Sorting needs to compare and move many items, so more rows mean more work and more time.

Interview Connect

Understanding how sorting scales helps you explain how data operations behave in real projects, showing you know how to handle growing data efficiently.

Self-Check

"What if the index was already sorted? How would the time complexity change when using sort_index()?"

Practice

1. What does the sort_index() function do in pandas?

easy

A. It filters rows based on a condition.

B. It sorts the DataFrame by its values.

C. It sorts the DataFrame by its index labels.

D. It resets the index to default integers.

Solution

1. Step 1: Understand the purpose of sort_index()

   This function organizes the DataFrame by sorting its row or column labels (index), not the data values.

2. Step 2: Compare options

   Only It sorts the DataFrame by its index labels. correctly describes sorting by index labels. Options B, C, and D describe other functions.

3. Final Answer:

   It sorts the DataFrame by its index labels. -> Option C

4. Quick Check:

   sort_index() = sort by index [OK]

Hint: Remember: sort_index() sorts labels, not data values [OK]

Common Mistakes:

• Confusing sort_index() with sort_values()
• Thinking it filters rows
• Assuming it resets the index

2. Which of the following is the correct syntax to sort a DataFrame df by its index in descending order?

easy

A. df.sort_index(order='desc')

B. df.sort_index(ascending=False)

C. df.sort_index(descending=True)

D. df.sort_index(reverse=True)

Solution

1. Step 1: Recall the parameter for sorting order

   The correct parameter is ascending, which accepts True or False.

2. Step 2: Check each option

   Only df.sort_index(ascending=False) uses ascending=False correctly. Options A, B, and D use invalid parameters.

3. Final Answer:

   df.sort_index(ascending=False) -> Option B

4. Quick Check:

   Use ascending=False for descending sort [OK]

Hint: Use ascending=False to sort index descending [OK]

Common Mistakes:

• Using invalid parameter names like descending or reverse
• Confusing ascending with descending
• Missing parentheses after sort_index

3. Given the DataFrame df with index labels [3, 1, 2] and values ["a", "b", "c"], what will df.sort_index() return as the index?

medium

A. [1, 2, 3]

B. [3, 2, 1]

C. ["a", "b", "c"]

D. [2, 3, 1]

Solution

1. Step 1: Understand the initial index

   The DataFrame index is [3, 1, 2], unordered.

2. Step 2: Apply sort_index()

   This sorts the index labels in ascending order: [1, 2, 3].

3. Final Answer:

   [1, 2, 3] -> Option A

4. Quick Check:

   sort_index() sorts index ascending by default [OK]

Hint: sort_index() sorts index ascending by default [OK]

Common Mistakes:

• Confusing index labels with values
• Expecting descending order without specifying
• Thinking values get sorted instead of index

4. What is wrong with this code snippet to sort the columns of DataFrame df by their labels?

df.sort_index(axis=1, inplace=False)

medium

A. It does not modify df because inplace=False is used.

B. The axis parameter should be 0 to sort columns.

C. sort_index() cannot sort columns, only rows.

D. inplace must be True to return a sorted DataFrame.

Solution

1. Step 1: Understand inplace parameter

   With inplace=False, the original DataFrame df is not changed; a sorted copy is returned.

2. Step 2: Check axis parameter

   axis=1 correctly specifies sorting columns, so no error there.

3. Step 3: Identify the issue

   Since the returned sorted DataFrame is not assigned, df remains unsorted.

4. Final Answer:

   It does not modify df because inplace=False is used. -> Option A

5. Quick Check:

   inplace=False returns new DataFrame, original unchanged [OK]

Hint: Use inplace=True or assign result to keep changes [OK]

Common Mistakes:

• Assuming inplace=False modifies original DataFrame
• Confusing axis=0 and axis=1 for columns
• Thinking sort_index() can't sort columns

5. You have a DataFrame df with a MultiIndex on rows. How can you sort the DataFrame by the second level of the index using sort_index()?

hard

A. df.sort_index(index=1)

B. df.sort_index(axis=1, level=1)

C. df.sort_index(by=1)

D. df.sort_index(level=1)

Solution

1. Step 1: Understand MultiIndex sorting

   To sort by a specific level in a MultiIndex, use the level parameter with the level number.

2. Step 2: Check each option

   df.sort_index(level=1) correctly uses level=1 to sort by the second index level. Options A, B, and C use invalid parameters.

3. Final Answer:

   df.sort_index(level=1) -> Option D

4. Quick Check:

   Use level parameter to sort MultiIndex levels [OK]

Hint: Use level=1 to sort by second index level [OK]

Common Mistakes:

• Using axis=1 to sort rows
• Using invalid parameters like by or index
• Not specifying level for MultiIndex sorting