Colorir células no Jogo da Vida do Python com base na idade das células

No desenvolvimento de simulações com NumPy e matplotlib, uma dúvida comum é como representar visualmente a evolução de cada célula ao longo do tempo.

No meu código, consigo definir a cor de fundo e das células, mas a ideia de colorir as células com base na idade delas ainda é um desafio. Essa visualização pode ajudar bastante na análise do comportamento de sistemas dinâmicos e na identificação de padrões emergentes.

Ao invés de usar a quantidade de vizinhos, que é o padrão, focar na idade das células oferece uma perspectiva mais interessante para entender a estabilidade e o ciclo de vida das estruturas formadas. Sem esse critério, a solução pode parecer simples no começo e cara no suporte. O valor aparece melhor quando operação, produto e engenharia olham para o mesmo risco.

Quem já tentou algo parecido ou tem alguma dica de como otimizar essa visualização? Acho que uma abordagem usando colormap dinâmico poderia ajudar bastante para deixar o feedback visual mais intuitivo. O valor aparece melhor quando operação, produto e engenharia olham para o mesmo risco. Por isso, o recorte precisa considerar manutenção, validação e caminho de volta. Esse contexto ajuda a separar ganho real de novidade difícil de sustentar.