In questo modo possiamo distinguere zone, della stessa area in esame, con un coefficiente di diffusione differente. Sia una matrice diagonale delle stesse dimensioni di , i suoi elementi nella diagonale esprimono i coefficienti di diffusione in ogni punto della regione. Si può osservare che i coefficienti di diffusione restano invariati nel tempo, quindi possono essere portati fuori dalla matrice . Siano e i coefficienti e senza il coefficiente di diffusione. Sostituendo nella matrice i coefficienti e al posto di e , la matrice così ottenuta è la . Ponendo . Sia la matrice ottenuta da sostituendo la matrice con e con , ora l' equazione () diviene