numpy.load() in Python

numpy.load() function return the input array from a disk file with npy extension(.npy).

Syntax : numpy.load(file, mmap_mode=None, allow_pickle=True, fix_imports=True, encoding=’ASCII’)

Parameters:
file : : file-like object, string, or pathlib.Path.The file to read. File-like objects must support the seek() and read() methods.
mmap_mode : If not None, then memory-map the file, using the given mode (see numpy.memmap for a detailed
description of the modes).
allow_pickle : Allow loading pickled object arrays stored in npy files.
fix_imports : Only useful when loading Python 2 generated pickled files on Python 3,which includes npy/npz files containing object arrays.
encoding : Only useful when loading Python 2 generated pickled files in Python 3, which includes npy/npz files containing object arrays.

Returns : Data stored in the file. For .npz files, the returned instance of NpzFile class must be closed to avoid leaking file descriptors.

Code #1 : Working

# Python program explaining
# load() function

import numpy as geek

a = geek.array(([i + j for i in range(3)
                       for j in range(3)]))
# a is printed.
print("a is:")
print(a)

geek.save('geekfile', a)
print("the array is saved in the file geekfile.npy")

# the array is saved in the file geekfile.npy
b = geek.load('geekfile.npy')

# the array is loaded into b
print("b is:")
print(b)

# b is printed from geekfile.npy
print("b is printed from geekfile.npy")

Output :

a is:
[0, 1, 2, 1, 2, 3, 2, 3, 4]
the array is saved in the file geekfile.npy
b is:
[0, 1, 2, 1, 2, 3, 2, 3, 4]
b is printed from geekfile.npy

Code #2:

# Python program explaining
# load() function

import numpy as geek

# a and b are numpy arrays.
a = geek.array(([i + j for i in range(3)
                       for j in range(3)]))
b = geek.array([i + 1 for i in range(3)])

# a and b are printed.
print("a is:")
print(a)
print("b is:")
print(b)

# a and b are stored in geekfile.npz
geek.savez('geekfile.npz', a = a, b = b)

print("a and b are stored in geekfile.npz")

# compressed file is loaded
c = geek.load('geekfile.npz')

print("after loading...")
print("a is:", c['a'])
print("b is:", c['b'])

Output :

a is:
[0 1 2 1 2 3 2 3 4]
b is:
[1 2 3]
a and b are stored in geekfile.npz
after loading...
a is: [0 1 2 1 2 3 2 3 4]
b is: [1 2 3]