No se esperaba que Yi entrara en el Draft de la NBA hasta el año 2009, dado que las autoridades chinas no autorizan a sus jugadores abandonar el país antes de los 22 años, pero a pesar de ello su equipo anunció que se presentaría al del 2007, donde fue elegido por Milwaukee Bucks en la sexta posición de la primera ronda. En su primer año jugó 25 minutos por partido, promediando 8.6 puntos y 5.2 rebotes por noche. Jugó además el Rookie Challenge del All-Star Weekend.