Skip to main content

Introduction to NumPy Arrays

In this tutorial, we'll delve into the fundamental concepts of arrays in NumPy, a powerful library for numerical computing in Python. Arrays are the central data structure in NumPy, providing a versatile and efficient way to work with large datasets.

What are Arrays?​

Arrays in NumPy are grid-like structures containing values of the same data type. They encapsulate raw data, facilitate element location, and interpretation. Here are some key points about arrays:

  • Data Structure: Arrays are the primary data structure in NumPy.
  • Homogeneous Data: All elements in an array are of the same type, known as the array's data type.
  • Indexing: Arrays can be indexed using non-negative integers, booleans, other arrays, or integers.
  • Rank: The rank of an array refers to the number of dimensions it possesses.
  • Shape: The shape of an array is a tuple indicating the size of each dimension.

Let's dive into some code examples to understand these concepts better.

Creating NumPy Arrays​

You can create NumPy arrays from Python lists using the numpy.array() function. For instance:

import numpy as np

# Creating a 1D array from a Python list
arr_1d = np.array([1, 2, 3, 4, 5])

# Creating a 2D array from a nested Python list
nested_list = [[1, 2, 3], [4, 5, 6]]
arr_2d = np.array(nested_list)

print("1D Array:")
print(arr_1d)
print("2D Array:")
print(arr_2d)

Accessing Array Elements​

You can access elements of a NumPy array using regular square bracket notation. For example:

# Accessing elements of a 2D array
print("First element of 2D array:", arr_2d[0])
print("Element at index (1,1) of 2D array:", arr_2d[1][1])
print("Last element of 2D array:", arr_2d[-1])

Common Terms​

When working with NumPy, you'll encounter several terms like ndarray, vectors, matrices, and tensors. Here's a brief overview:

  • ndarray: Short for n-dimensional array, representing arrays with any number of dimensions.
  • Vector: An array with a single dimension.
  • Matrix: An array with two dimensions.
  • Tensor: An array with three or more dimensions.

Understanding Array Axes and Shapes​

Arrays have axes corresponding to their dimensions. The shape of an array indicates its size along each axis. Here's how you can access this information:

# Getting the shape of a 2D array
print("Shape of 2D array:", arr_2d.shape)

# Getting the data type of elements in the array
print("Data type of array:", arr_2d.dtype)

Numpy Basics Tutorial​

In this tutorial, we will cover the basics of numpy arrays, including creating arrays with different methods and specifying data types.

Importing Numpy​

Firstly, you need to import numpy. It's a common convention to alias numpy as np:

import numpy as np

Creating Numpy Arrays​

To create a numpy array, you can pass a Python list into np.array(). Let's start with a simple example:

arr = np.array([1, 2, 3, 4])
print(arr)

This will output:

[1 2 3 4]

Specifying Data Types​

You can specify the data type of the elements in the array using the dtype parameter. For example:

arr_str = np.array([1, 2, 3, 4], dtype=str)
arr_int8 = np.array([1, 2, 3, 4], dtype=np.int8)
print(arr_str)
print(arr_int8)

Creating Arrays with Zeros, Ones, and Empty​

Numpy provides convenient methods to create arrays filled with zeros, ones, or random values using np.zeros(), np.ones(), and np.empty() respectively.

zeros_arr = np.zeros(5)
ones_arr = np.ones(5)
empty_arr = np.empty(5)

print(zeros_arr)
print(ones_arr)
print(empty_arr)

Creating a Range​

You can create a range of numbers using np.arange(). It works similarly to Python's built-in range() function.

range_arr = np.arange(5)
print(range_arr)

# Specifying start, stop, and step
custom_range_arr = np.arange(2, 10, 2)
print(custom_range_arr)

Creating Linearly Spaced Values​

Numpy's np.linspace() function generates evenly spaced numbers over a specified interval.

lin_space_arr = np.linspace(0, 10, 5)
print(lin_space_arr)

# Splitting into 3 parts
lin_space_3_arr = np.linspace(0, 10, 3)
print(lin_space_3_arr)

# Splitting into 20 parts
lin_space_20_arr = np.linspace(0, 10, 20)
print(lin_space_20_arr)

Conclusion​

In this tutorial, we've covered the basics of creating numpy arrays using various methods and specifying data types. Numpy provides powerful tools for numerical computing in Python, and understanding these fundamental concepts will help you work more effectively with arrays and matrices in your data science projects.