Colorir células no Jogo da Vida de Conway com base na idade ou vizinhos

No desenvolvimento de simulações visuais, uma dúvida comum é como representar variáveis dinâmicas, como a idade das células, usando cores diferentes. No meu projeto, implementei o Jogo da Vida em NumPy e Matplotlib, e a ideia era fazer as células mudarem de cor conforme o número de vizinhos ou a idade. Apesar de conseguir definir cores de fundo e células, a visualização fica meio plana, sem a gradação que eu queria.

A questão é: como fazer a coloração das células refletir atributos como idade ou quantidade de vizinhos, sem ficar tudo numa mesma cor? Acredito que essa visualização ajuda bastante na análise do comportamento das células ao longo do tempo, além de tornar o código mais didático.

No seu entendimento, qual a melhor abordagem pra representar visualmente esses atributos de forma clara e eficiente? Acho que vale discutir também o impacto na performance, já que mudanças de cor podem exigir atualizações mais frequentes na tela. Sem esse critério, a solução pode parecer simples no começo e cara no suporte.