Stampa div in base al pulsante stampa cliccato

[Mich_]

<div id="spazio1">...</div><button onclick="stampa(1);">
<div id="spazio2">...</div><button onclick="stampa(2);">
<div id="spazio3">...</div><button onclick="stampa(3);">

JS:

function stampa(n) {
  for (var i=1; i<=3; i++) {
    document.getElementById('spazio'+i).style.display = (i==n) ? 'block' : 'none';
  }
  document.print();
}

Spero di non aver fatto errori.
Sposto comunque nel forum JS.

Poi occorre vedere cosa deve fare il browser dopo la stampa: con questo codice non ripristina la pagina originale.