Essential cookies keep authentication working. With your permission, we also use analytics cookies to understand and improve the product. Read our Privacy Policy

DataEngPrep.tech
QuestionsPracticeAI CoachDashboardPricingBlog
ProLogin
Home/Questions/Python/Coding/Optimize a function to calculate moving averages of user engagement.

Optimize a function to calculate moving averages of user engagement.

Python/Codinghard0.5 min read

Reviewed by Aditya Kumar · Last reviewed 2026-03-24

**Why Moving Averages in Engagement:** DAU/MAU curves, session duration trends, retention metrics—all use rolling windows. Naive O(n*window) per row is unacceptable at scale. **Optimization:** (1) Cumulative sum: precompute cumsum, then avg[i] = (cumsum[i] - cumsum[i-w]) / w....

🤖 Analyze Your Answer
Frequency
Low
Asked at 1 company
Category
179
questions in Python/Coding
Difficulty Split
127E|24M|28H
in this category
Total Bank
1,863
across 7 categories
Asked at these companies
Disney+ Hotstar
Key Concepts Tested
optimizationsparkwindow

Why This Question Matters

This hard-level Python/Coding question appears frequently in data engineering interviews at companies like Disney+ Hotstar. While less common, it tests deeper understanding that distinguishes strong candidates. Mastering the underlying concepts (optimization, spark, window) will help you answer variations of this question confidently.

How to Approach This

This is a senior-level question that tests architectural thinking. Lead with the high-level design, then drill into specifics. Discuss trade-offs explicitly - there is rarely one correct answer. Show awareness of scale, fault tolerance, and operational complexity.

Expert Answer
105 words

Why Moving Averages in Engagement: DAU/MAU curves, session duration trends, retention metrics—all use rolling windows. Naive O(n*window) per row is unacceptable at scale.

Optimization: (1) Cumulative sum: precompute cumsum, then avg[i] = (cumsum[i] - cumsum[i-w]) / w. O(n) total. (2) Pandas: rolling(window=7).mean()—vectorized, uses underlying C. (3) Streaming: maintain deque of size w, running sum—O(1) per update. (4) Exponential: EMA = αnew + (1-α)prev—O(1), no window storage.

Cost: At Hotstar, 10M users * 30 days with window=7: naive = 2.1B ops; cumsum = 300M. Use Spark window functions for distributed: rowsBetween(-6, 0).

def opt_moving_avg(arr, w):
cumsum = np.cumsum(np.insert(arr, 0, 0))
return (cumsum[w:] - cumsum[:-w]) / w

⚡
Pro Tip

Red Flag: Python loop for each window. Pro-Move: 'We use Pandas rolling for batch; Redis + deque for real-time dashboards—sub-second latency for 7-day MA.'

Want all answers as a PDF for offline study?
Seven focused volumes with 750+ in-depth answers — Answer Vault →

Related Python/Coding Questions

easyWhat are traits in Scala, and how are they different from classes?FreemediumWrite a Python function to check if a string is a palindrome.FreeeasyWhat is the difference between a list and a tuple in Python?FreeeasyExplain the difference between shallow copy and deep copy in Python.FreeeasyWrite a Python function to find the first non-repeating character in a string.Free

Level up your prep

Recommended
Educative
Educative Unlimited

800+ hands-on courses — Grokking System Design, Coding Patterns, and AI mock interviews for your DE loop.

Start learning →

Some links below are affiliate links. If you buy through them we may earn a small commission at no extra cost to you — it helps keep DataEngPrep free.

According to DataEngPrep.tech, this is one of the most frequently asked Python/Coding interview questions, reported at 1 company. DataEngPrep.tech maintains an editor-reviewed database of 1,863 data engineering interview questions across 7 categories.

← Back to all questionsMore Python/Coding questions →
Categories
All QuestionsSQLSpark / Big DataPython / CodingSystem DesignCloud / ToolsBehavioral
By Company
AmazonGoogleDatabricksSnowflakeAWSAzureMicrosoftNetflixUberTCS
Interview Guides
All GuidesTop SQL QuestionsTop Spark QuestionsPySpark QuestionsTop Python QuestionsTop System DesignKafka QuestionsAirflow QuestionsSQL Window FunctionsETL QuestionsData Modeling
Products
AI Interview CoachAnswer AnalyzerSQL PlaygroundResume AnalyzerAnswer Vault PDFsPricing
Company
About & Editorial PolicyContact UsAI DisclosureDisclaimerTerms of ServicePrivacy Policy
© 2026 DataEngPrep.tech. All rights reserved.
AboutBlogContactDisclaimer