diff --git a/doc/scripts/eval_performance.py b/doc/scripts/eval_performance.py
index f6087e02a9330..559689ef5915b 100644
--- a/doc/scripts/eval_performance.py
+++ b/doc/scripts/eval_performance.py
@@ -6,6 +6,7 @@
 from pandas import DataFrame
 
 setup_common = """from pandas import DataFrame
+import numpy as np
 df = DataFrame(np.random.randn(%d, 3), columns=list('abc'))
 %s"""