I would think it likely that those arrests are designed to put pressure on the US to release the Iranian officials who have been detained by the US authorities since January (against the wishes of the Iraq government).

Two wrongs don't make a right of course - and in any case I imagine that it's an ineffective way of exerting pressure, since it probably rather suits the US authorities to have Iran do this.