Make pyamg example run

Author: stefanv

advanced/scipy_sparse/examples/pyamg_with_lobpcg.py

@@ -7,6 +7,7 @@
 Dirichlet boundary conditions.
 """
 
+import numpy as np
 import scipy as sp
 import matplotlib.pyplot as plt
 
@@ -21,7 +22,7 @@
 ml = smoothed_aggregation_solver(A)
 
 # initial approximation to the K eigenvectors
-X = sp.rand(A.shape[0], K)
+X = np.random.random((A.shape[0], K))
 
 # preconditioner based on ml
 M = ml.aspreconditioner()