DOC: fix missing import & clarify python doc example

2 jobs for doc-numpy-pipeline in 4 minutes and 5 seconds (queued for 2 seconds)
Status Name Job ID Coverage
  Precheck
passed fast-build #6557

00:04:05

 
  External
passed cdash:ubuntu-18.04-fast