如何使用javascript键代码切换div

<div id="div1">内容1</div>
<div id="div2">内容2</div>
<button onclick="toggleDiv()">切换</button>

function toggleDiv() {
  var div1 = document.getElementById("div1");
  var div2 = document.getElementById("div2");
  
  if (div1.style.display === "none") {
    div1.style.display = "block";
    div2.style.display = "none";
  } else {
    div1.style.display = "none";
    div2.style.display = "block";
  }
}