You want to turn the Wi-Fi function off at the same time the Walkman screen is turned off.
- Swipe the status bar down, tap
- 
[WI-FI] - 
- [Advanced] - [Keep Wi-Fi on during sleep] - desired settings.
- [WI-FI] - - [Advanced] - [Keep Wi-Fi on during sleep] - desired settings.