Loading interview...

The Column Shuffle

A medium Python mock interview question on DataDriven. Practice with AI-powered feedback, real code execution, and a hire/no-hire decision.

Domain
Python
Difficulty
medium
Seniority
staff

Interview Prompt

The ML feature store requires one row per entity with amounts in separate columns. Given records with an ID and variable-length amounts, reshape each group so amounts become amount_1, amount_2, amount_3 in a single dictionary per ID.

How This Interview Works

  1. Read the vague prompt (just like a real interview)
  2. Ask clarifying questions to the AI interviewer
  3. Write your python solution with real code execution
  4. Get instant feedback and a hire/no-hire decision