Skip to content
Related Articles

Related Articles

Improve Article
numpy.load() in Python
  • Last Updated : 29 Nov, 2018

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]

 Attention geek! Strengthen your foundations with the Python Programming Foundation Course and learn the basics.  

To begin with, your interview preparations Enhance your Data Structures concepts with the Python DS Course. And to begin with your Machine Learning Journey, join the Machine Learning – Basic Level Course




My Personal Notes arrow_drop_up
Recommended Articles
Page :