The dynamics of waves and eddies in the upper ocean plays an important role in the climate variation of tropical and subtropical regions. Previous diagnoses for annual Rossby waves in oceanic model outputs manifested zonally alternating signals (ZASs) in the time-averaged distributions of wind input as well as pressure-flux divergence terms in the budget equation of wave energy. This is the case when the annual mean of the wind input is estimated as the inner product of simulated velocity vector and wind stress vector in previous studies. The present study proposes a new mathematical expression for estimating the wind input that is analogous to one derived from the quasi-geostrophic potential vorticity equation. Namely, the wind input is estimated as the negative of the product of pseudo-streamfunction and wind stress curl, the latter of which is associated with the horizontal divergence of Ekman velocity. This can be interpreted as replacement of kinetic energy input with gravitational potential energy input. Pseudo-streamfunction in the present study is inverted from Ertel’s potential vorticity anomaly and is seamlessly available at all latitudes. This contrasts with the quasi-geostrophic streamfunction which is singular at the equator. The new expression enables reducing ZASs in the horizontal distributions of both wind input and pressure-flux divergence terms, without harming the qualitative advantage of energy flux vectors to indicate the group velocity of waves at all latitudes.