Risolvere il problema della miscela matematica usando Python

Mi sono imbattuto in un problema di matematica per le miscele e stavo pensando a come spiegare la soluzione usando Python e grafici. Il problema è:

Ti viene dato un serbatoio del peso di 20 galloni di miscela di olio e acqua. La percentuale di olio è del 40%. L'acqua viene aggiunta per diluire la composizione al 25% di olio.
a) Qual è il peso totale della miscela diluita?
b) Quanta acqua è stata aggiunta?

Per risolvere il problema, scriviamo prima le note.

import numpy as np
import matplotlib.pyplot as plt

orig_mix_weight = 20       # 20 gallons of original mixture
orig_mix_oil_per = 0.4     # percentage of oil in the original mixture
diluted_mix_oil_per = 0.25 # percentage of oil in the diluted mixture

# percentage of water in the original mixture
orig_mix_water_per = 1 - orig_mix_oil_per
orig_mix_water_per
0.6

water, oil = 10, 9  # Assume 10 represents water and 9 oil in the matrix

# create an array with water and oil as elements in the proportion
orig_mix = np.random.choice([water, oil],
                           size=(orig_mix_weight,),
                           p=[orig_mix_water_per, orig_mix_oil_per]) # probability
orig_mix = orig_mix.reshape(5, 4) # make it a 5x4 matrix for visualization
orig_mix

      
                

                    
                    
                
            

[m,n] = np.shape(orig_mix)

plt.figure()
plt.imshow(orig_mix, alpha=0.8, cmap='Reds')
plt.xticks(np.arange(n))
plt.yticks(np.arange(m))
plt.title('Original Solution')
plt.show()

      
                

                    
                    
                
            

orig_mix_oil_weight = orig_mix_weight * orig_mix_oil_per
orig_mix_oil_weight # weight of oil in original mixture
8.0
diluted_mix_oil_weight = orig_mix_oil_weight
diluted_mix_oil_weight # weight of oil in diluted mixture
8.0

diluted_mix_alcohol_per = 0.25
diluted_mix_weight = 1 / diluted_mix_oil_per * diluted_mix_oil_weight
diluted_mix_weight
32.0

diluted_mix_water_per = 1- diluted_mix_alcohol_per

diluted_mix = np.random.choice([water, oil],
                           size=(int(diluted_mix_weight),),
                               p=[diluted_mix_water_per, diluted_mix_alcohol_per])
diluted_mix = diluted_mix.reshape(8, 4) # make it a 5x4 matrix for visualization

[m,n] = np.shape(diluted_mix)

plt.figure()
plt.imshow(diluted_mix, alpha=0.8, cmap='Reds')
plt.xticks(np.arange(n))
plt.yticks(np.arange(m))
plt.title('Diluted Solution')
plt.show()

      
                

                    
                    
                
            

diluted_mix_weight - orig_mix_weight
12.0