# Given nums = [3, 7, 2, 8, 5], use a list comprehension with the walrus operator to capture only elem

Canonical URL: <https://datadriven.io/problems/given-nums-3-7-2-8-5-use-a-list-comprehension-with-179a3402>

Domain: Python · Difficulty: medium

## Problem

Given nums = [3, 7, 2, 8, 5], use a list comprehension with the walrus operator to capture only elements whose square exceeds 20, collecting their squares. Print the list.

## Related

- [All practice problems](https://datadriven.io/problems)
- [Mock interview mode](https://datadriven.io/interview/given-nums-3-7-2-8-5-use-a-list-comprehension-with-179a3402)
- [Python Interview Questions](https://datadriven.io/python-interview-questions)
- [Data Engineering Interview Prep Guide](https://datadriven.io/data-engineer-interview-prep)
- [Daily Challenge](https://datadriven.io/daily)

---

Source: DataDriven (https://datadriven.io). 100% free data engineering interview prep. Live code execution against Postgres 16, Python 3.11, and Spark sandboxes. No paywall, no premium tier, no signup gate.