The answer is yes, that you can get an analogous theorem, because $\mathbb{Z}_{p^k}$ is a local ring.
The equation $PAP^{-1} = B$ can be rewritten $PA = BP$. Write $A = \text{diag}(a), B = \text{diag}(b), a = \{a_i\}, b = \{b_j\}$. Then this implies that for any $i, j$, we have that $P_{i, j} a_i = P_{i, j} b_j$.
As $P \in GL_n(\mathbb{Z}_{p^k})$, we know that $\text{det}(P)$ must be a unit. We can write $\text{det}(P) = \sum_{\sigma \in S_n} (-1)^{\text{sgn}(\sigma)} \prod_i P_{i, \sigma(i)}$. As $\mathbb{Z}_{p^k}$ is a local ring, the set of nonunits is an additively closed, so at least one of the $\prod_i P_{i, \sigma(i)}$ must be a unit. Correspondingly, there must be some $\sigma$ such that $P_{i, \sigma(i)}$ is a unit for all $i$.
Then $P_{i, \sigma(i)} a_i = P_{i, \sigma(i)} b_{\sigma(i)}$. But $P_{i, \sigma(i)}$ is a unit for each $i$ - so we can cancel it to get $a_i = b_{\sigma(i)}$ for each $i$. Therefore, $\sigma$ is a permutation of diagonal entries that takes $A$ to $B$, and we are done.