The idea was to give a self-contained example that a user can simply copy-paste... Which is not the case without the import statement. (How to call `np.array` later without this line?) Still, no problem removing it.