Adding and removing columns in R Programming - Time & Space Complexity

Time Complexity: Adding and removing columns

O(n)

Understanding Time Complexity

When we add or remove columns in a data frame, the time it takes depends on how many columns we handle.

We want to know how the work grows as the number of columns changes.

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

# Create a data frame with n columns
n <- 1000
df <- data.frame(matrix(1:(10*n), nrow=10, ncol=n))

# Add a new column
 df$new_col <- 1:10

# Remove a column
 df$new_col <- NULL
    

This code creates a data frame with many columns, adds one new column, then removes it.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Adding or removing a column involves copying or updating the data frame's column structure.
• How many times: Each add or remove touches all rows but only once per operation.

How Execution Grows With Input

Adding or removing a column means working with all rows for that column, but the number of columns affects how the data frame is managed internally.

Input Size (n columns)	Approx. Operations
10	Work on 10 rows x 10 columns plus 10 rows for new column
100	Work on 10 rows x 100 columns plus 10 rows for new column
1000	Work on 10 rows x 1000 columns plus 10 rows for new column

Pattern observation: The work grows roughly linearly with the number of columns because the data frame structure updates depend on column count.

Final Time Complexity

Time Complexity: O(n)

This means the time to add or remove a column grows in a straight line as the number of columns increases.

Common Mistake

[X] Wrong: "Adding or removing a column is instant and does not depend on the number of columns."

[OK] Correct: The data frame must update its structure, which involves handling all columns internally, so more columns mean more work.

Interview Connect

Understanding how data frame operations scale helps you write efficient code and explain your choices clearly in real projects.

Self-Check

"What if we added multiple columns at once instead of one by one? How would the time complexity change?"

Practice

1. In R, how do you add a new column named age with values c(25, 30, 35) to an existing dataframe df?

easy

A. df <- c(df, age = c(25, 30, 35))

B. df[age] = c(25, 30, 35)

C. df$age <- c(25, 30, 35)

D. df$age = NULL

Solution

1. Step 1: Understand how to add columns in R

   In R, you add a new column to a dataframe by assigning a vector to df$column_name.

2. Step 2: Apply the assignment for the new column

   Assigning c(25, 30, 35) to df$age adds the new column age with those values.

3. Final Answer:

   df$age <- c(25, 30, 35) -> Option C

4. Quick Check:

   Use $ and assign vector to add column [OK]

Hint: Use df$column <- values to add columns fast [OK]

Common Mistakes:

• Using NULL to add a column instead of removing
• Trying to add column with c() outside assignment
• Using incorrect dataframe syntax like c(df, ...)

2. Which of the following is the correct syntax to remove the column height from dataframe df in R?

easy

A. df$height <- NULL

B. df$height = c()

C. df$height <- NA

D. remove(df$height)

Solution

1. Step 1: Recall how to remove columns in R

   Assigning NULL to a dataframe column removes it completely.

2. Step 2: Apply removal syntax

   Setting df$height <- NULL deletes the height column from df.

3. Final Answer:

   df$height <- NULL -> Option A

4. Quick Check:

   Assign NULL to remove column [OK]

Hint: Remove columns by assigning NULL: df$col <- NULL [OK]

Common Mistakes:

• Assigning empty vector c() instead of NULL
• Assigning NA which keeps column but fills with missing
• Using remove() which is not for dataframe columns

3. What will be the output of the following R code?

df <- data.frame(name = c("Ann", "Bob"), score = c(90, 85))
df$passed <- c(TRUE, FALSE)
df$score <- NULL
df

medium

A. A dataframe with columns: name, passed

B. A dataframe with columns: name, score, passed

C. A dataframe with columns: name, score

D. Error because score column cannot be removed

Solution

1. Step 1: Add new column 'passed'

   The code adds a logical column passed with values TRUE and FALSE.

2. Step 2: Remove the 'score' column

   Assigning NULL to df$score deletes the score column.

3. Final Answer:

   A dataframe with columns: name, passed -> Option A

4. Quick Check:

   Added passed, removed score, left name [OK]

Hint: NULL removes columns; new assignment adds columns [OK]

Common Mistakes:

• Thinking score column remains after NULL assignment
• Expecting error on removing existing column
• Confusing TRUE/FALSE vector with numeric

4. Consider this R code snippet:

df <- data.frame(x = 1:3, y = 4:6)
df$z = c(7, 8)
df$y <- NULL
df

What is the problem and how to fix it?

medium

A. Error because data.frame cannot have NULL columns; fix by using list

B. No error; code runs fine and removes y column

C. Error because NULL cannot remove columns; fix by using NA

D. Warning because length of z vector does not match rows; fix by matching length

Solution

1. Step 1: Check length of new column vector

   The dataframe has 3 rows, but c(7, 8) has length 2, causing a length mismatch warning.

2. Step 2: Fix by providing vector of correct length

   Use c(7, 8, 9) or recycle values properly to match 3 rows.

3. Final Answer:

   Warning due to length mismatch; fix by matching vector length -> Option D

4. Quick Check:

   New column vector length must match dataframe rows [OK]

Hint: New column vector length must equal dataframe rows [OK]

Common Mistakes:

• Ignoring vector length mismatch causing warning
• Thinking NULL cannot remove columns
• Trying to assign NA to remove columns

5. You have a dataframe df with columns id, score1, and score2. You want to create a new column average that is the mean of score1 and score2, then remove score1 and score2. Which code correctly does this?

hard

A. df$average <- mean(df$score1, df$score2)
df$score1 <- NULL
df$score2 <- NULL

B. df$average <- rowMeans(df[c("score1", "score2")])
df$score1 <- NULL
df$score2 <- NULL

C. df$average <- mean((df$score1 + df$score2) / 2)
df$score1 <- NULL
df$score2 <- NULL

D. df$average <- (score1 + score2) / 2
df$score1 <- NULL
df$score2 <- NULL

Solution

1. Step 1: Calculate average of two columns row-wise

   Use rowMeans() on the subset df[c("score1", "score2")] to get the mean for each row.

2. Step 2: Remove the original score columns

   Assign NULL to df$score1 and df$score2 to delete them.

3. Final Answer:

   df$average <- rowMeans(df[c("score1", "score2")]) df$score1 <- NULL df$score2 <- NULL -> Option B

4. Quick Check:

   Use rowMeans() for row-wise average, then remove columns [OK]

Hint: Use rowMeans() for row averages, then assign NULL to remove [OK]

Common Mistakes:

• Using mean() which computes overall mean, not row-wise
• Referencing columns without df$ prefix
• Trying to remove columns by assigning NA