Capillary imbibition in unsaturated rocks is important for the exploitation of tight reservoirs, such as oil and gas reservoirs. However, the physical properties of natural rocks tend to be relatively uneven, mainly in the heterogeneity of material composition and pore space. Reservoir heterogeneity is an important factor affecting the exploitation of oil fields and other reservoirs, which can be evaluated by the pore structure tortuosity fractal dimension DT of rock. The greater the value of DT, the stronger the heterogeneity of sandstone. Two types of sandstone with high and low permeability were selected to study the effect of heterogeneity on the imbibition behavior by using high-resolution X-ray imaging and neutron radiography. Quantitative results of the wetting front position for each specimen were extracted from the neutron images. The wetting front advanced linearly with the power index of time t1/(2DT). Different values of DT were selected to estimate and discuss the effect of the heterogeneity on sorptivity. A modified L-W equation was employed to predict the sorptivity. Comparing with the experimental results, the heterogeneity plays a significant role in determining the sorptivity. The modified model provides a reference for the prediction of the sorptivity of the same types of sandstones studied in this paper.