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 Job ID Name Coverage
  Precheck
passed #6557
fast-build

00:04:05

 
  External
passed cdash:ubuntu-18.04-fast