removed plotting of complex types

app.py

@@ -42,20 +42,24 @@ x_dash_down, y_dash_down = transform(x,-y)
 
 t = np.array([[a_00,a_01], [a_10, a_11]], dtype=np.float64)
 
-try:
-    evl, evec = np.linalg.eig(t)
-    fig, ax = plt.subplots()
-    ax.plot(x_dash_up,y_dash_up,'r')
-    ax.plot(x_dash_down,y_dash_down, 'g')
-    ax.quiver(0,0,evec[0][0],evec[0][1],scale=1,scale_units ='xy',angles='xy', facecolor='yellow', label='$\lambda_0$')
-    ax.quiver(0,0,evec[1][0],evec[1][1],scale=1,scale_units ='xy',angles='xy', facecolor='blue',label='$\lambda_1$')
-    ax.set_xlim(-5,5)
-    ax.set_ylim(-5,5)
-    ax.set_aspect('equal', adjustable='box')
-    fig.legend()
-    st.pyplot(fig)
-
-    df = pd.DataFrame({'Eigen Values': evl, 'Eigen Vectors': [str(evec[0]), str(evec[1])]})
-    st.table(df)
-except:
-    st.write("Given matrix has eigen vectors in complex space")
+
+evl, evec = np.linalg.eig(t)
+fig, ax = plt.subplots()
+ax.plot(x_dash_up,y_dash_up,'r')
+ax.plot(x_dash_down,y_dash_down, 'g')
+if not (np.iscomplex(evl).any() or np.iscomplex(evec).any()):
+    ax.quiver(0,0,evec[0,0]*evl[0],evec[1,0]*evl[0],scale=1,scale_units ='xy',angles='xy', facecolor='yellow', label='$eigen\ vector_{\lambda_0}$')
+    ax.quiver(0,0,evec[0,1]*evl[1],evec[1,1]*evl[1],scale=1,scale_units ='xy',angles='xy', facecolor='blue',label='$eigen\ vector_{\lambda_1}$')
+ax.set_xlim(-10,10)
+ax.set_ylim(-10,10)
+ax.set_aspect('equal', adjustable='box')
+fig.legend()
+st.pyplot(fig)
+
+df = pd.DataFrame({'Eigen Values': evl, 'Eigen Vectors': [str(evec[:,0]), str(evec[:,1])],\
+    'Transformed Eigen Vectors': [str(evec[:,0]*evl[0]), str(evec[:,1]*evl[1])]})
+st.table(df)
+
+if np.iscomplex(evl).any() or np.iscomplex(evec).any():
+    st.write("Due to complex eigen vectors or eigne values, the transformed eigen vectors are not\
+        displayed...")