Understanding community assembly mechanisms is helpful to predict community dynamics. To explore which community assembly mechanism(s) drive(s) the grassland restoration in semi-arid region, we investigated the relationships between plant trait and species relative abundance (SRA), and estimated community functional diversity indices for each community under different treatments (enclosure, grazing and mowing treatment) in a restoration region of Stipa grandis - Leymus chinensis communities in the northern China from 2010 to 2012. There was a high fraction of significant relationships between trait value and SRA, suggesting that niche theory structured the grassland restoration in this region. The functional richness was higher and the functional divergence was lower in the enclosure community than that in the grazing or mowing community, and significantly positive plant height - SRA relationship was found in the enclosure community. These findings demonstrated that limiting similarity based on niche theory was more important in structuring the enclosure community and that environmental filtering based on niche theory played a more important role in driving the grazing or mowing community. Only the factor of year significantly affected the functional evenness (FEve), and the lowest FEve in 2011 implied that the relatively lower precipitation could enhance the effect of limiting similarity on community assembly in the semi-arid grassland.