September 11, 2024

Python Tutorials

Python Arrays

In Python, an array is a collection of items stored at contiguous memory locations. While Python does not have a built-in array data type like some other programming languages, it provides various ways to work with arrays using libraries such as array and numpy. Arrays are used to store multiple values in a single variable, and they can be more efficient than lists when dealing with large amounts of numeric data.

1. Arrays with the array Module

The array module provides a way to create arrays of uniform data types. Unlike lists, which can hold items of different types, arrays created with the array module can only contain items of the same data type.

1.1. Creating an Array

To create an array, you first need to import the array module. The syntax for creating an array is:

import array

# Create an array of integers
arr = array.array('i', [1, 2, 3, 4, 5])

print("Array:", arr)

Here, 'i' indicates that the array will hold integers. Other type codes include 'f' for floats, 'd' for doubles, etc.

1.2. Accessing Array Elements

You can access elements in an array using indexing, similar to lists:

import array

arr = array.array('i', [1, 2, 3, 4, 5])

# Accessing elements
print("First element:", arr[0])
print("Last element:", arr[-1])

1.3. Modifying Array Elements

Array elements can be modified by assigning new values to specific indices:

import array

arr = array.array('i', [1, 2, 3, 4, 5])

# Modify the second element
arr[1] = 10

print("Modified Array:", arr)

1.4. Adding and Removing Elements

  • arr.append(x): Adds an element x to the end of the array.
  • arr.insert(i, x): Inserts an element x at position i.
  • arr.pop(i): Removes and returns the element at position i. If i is not specified, the last element is removed.
  • arr.remove(x): Removes the first occurrence of the element x from the array.
import array

arr = array.array('i', [1, 2, 3, 4, 5])

# Append an element
arr.append(6)

# Insert an element at the second position
arr.insert(1, 15)

# Remove an element
arr.remove(4)

print("Array after modifications:", arr)

1.5. Array Methods

  • arr.index(x): Returns the index of the first occurrence of x in the array.
  • arr.count(x): Returns the number of occurrences of x in the array.
  • arr.reverse(): Reverses the elements of the array in place.
  • arr.extend(iterable): Appends items from an iterable to the end of the array.
import array

arr = array.array('i', [1, 2, 3, 2, 4, 2])

# Count occurrences of 2
print("Count of 2:", arr.count(2))

# Find index of the first occurrence of 4
print("Index of 4:", arr.index(4))

# Reverse the array
arr.reverse()
print("Reversed Array:", arr)

2. Arrays with the numpy Library

numpy is a powerful library for numerical computing in Python. It provides support for large, multi-dimensional arrays and matrices, along with a collection of mathematical functions to operate on these arrays.

2.1. Creating a NumPy Array

To use numpy, you need to install it first (if it’s not already installed) and then import it:

pip install numpy
import numpy as np

# Create a NumPy array
arr = np.array([1, 2, 3, 4, 5])

print("NumPy Array:", arr)

2.2. Advantages of NumPy Arrays

NumPy arrays are more efficient than Python lists for large datasets because they provide:

  • Better performance: NumPy arrays use less memory and are faster than lists, especially for large datasets.
  • Convenient methods: NumPy provides a wide range of functions for performing mathematical operations on arrays.
  • Multi-dimensional arrays: NumPy supports multi-dimensional arrays, which are essential for complex data structures like matrices.

2.3. Basic Operations with NumPy Arrays

import numpy as np

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

# Element-wise operations
arr = arr + 2
print("Array after adding 2:", arr)

arr = arr * 3
print("Array after multiplying by 3:", arr)

# Square each element
arr = arr ** 2
print("Array after squaring:", arr)

2.4. Multi-Dimensional Arrays

NumPy allows you to create multi-dimensional arrays (matrices):

import numpy as np

# Create a 2x3 matrix
matrix = np.array([[1, 2, 3], [4, 5, 6]])

print("Matrix:\n", matrix)

# Access an element
print("Element at (1, 2):", matrix[1, 2])

# Perform matrix operations
transposed = np.transpose(matrix)
print("Transposed Matrix:\n", transposed)

You May Like