Open In App

numpy.load() in Python

Last Updated : 29 Nov, 2018
Comments
Improve
Suggest changes
Like Article
Like
Report
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 Python3
# 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: Python3
# 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]

Next Article
Practice Tags :

Similar Reads