Para el acceso de haz en 3D en sistemas masivos de entrada múltiple y salida múltiple (MIMO), este documento propone un diseño de libro de códigos 3D de dos etapas factible y un algoritmo de entrenamiento de haz rápido correspondiente. El libro de códigos completo consta de un libro de códigos primario y un libro de códigos auxiliar: el libro de códigos primario produce un haz direccional básico utilizando un menor número de cambios de fase para mantener una menor complejidad de hardware. El pequeño libro de códigos auxiliar crea haces más finos centrados en cada haz principal. Además, se propone un esquema de búsqueda de haces para el diseño de libro de códigos de dos niveles. La búsqueda del haz principal se realiza utilizando un algoritmo de búsqueda binaria modificado basado en el umbral de relación señal-ruido; con el libro de códigos auxiliar refinado, la búsqueda del haz auxiliar se completa utilizando una búsqueda binaria básica. Los resultados de la simulación verifican las ventajas de rendimiento de nuestro libro de códigos propuesto en términos de ganancia de formación de haz y precisión de acceso de haz en sistemas MIMO. Además, la comparación de la complejidad de entrenamiento de haz muestra que nuestro esquema de búsqueda de haz de dos etapas especialmente diseñado tiene ventajas significativas sobre otras soluciones existentes.