Sure. In Settings > General > Cellular > disable Cellular Data

This will restrict all data to Wi-Fi

enter image description here

For more information, see iOS: About the Cellular Data setting from Apple Support


Settings > General > Cellular > Cellular Data > Off.