Data Structures for Data Science

Data structures are crucial for efficiently handling and analyzing data in data science. Python provides several built-in and external data structures, such as lists, dictionaries, and NumPy arrays, that are widely used in data science tasks.

In this lesson, you’ll learn about key data structures for data science and their applications.

Lists

Lists are mutable sequences in Python that can hold a variety of data types. They are used for simple data storage and manipulation.

# Example: Working with lists
my_list = [1, 2, 3, 4]
my_list.append(5)
print(my_list)  # Output: [1, 2, 3, 4, 5]

Dictionaries

Dictionaries store data as key-value pairs, making them ideal for lookups and mapping relationships.

# Example: Using dictionaries
data = {'Name': 'Alice', 'Age': 25}
print(data['Name'])  # Output: Alice

Sets

Sets are unordered collections of unique items, useful for removing duplicates or performing set operations like union and intersection.

# Example: Working with sets
my_set = {1, 2, 3}
my_set.add(2)  # No duplicates allowed
print(my_set)  # Output: {1, 2, 3}

Tuples

Tuples are immutable sequences often used to store related data that shouldn’t change, such as coordinates or database records.

# Example: Using tuples
coordinates = (10, 20)
print(coordinates[0])  # Output: 10

NumPy Arrays

NumPy arrays are used for numerical computations and handling large datasets efficiently. They support vectorized operations, making them faster than lists for numerical tasks.

# Example: NumPy arrays
import numpy as np

arr = np.array([1, 2, 3])
print(arr * 2)  # Output: [2, 4, 6]

Pandas DataFrames

Pandas DataFrames are 2-dimensional labeled data structures, similar to tables. They are essential for data manipulation and analysis.

# Example: Pandas DataFrame
import pandas as pd

data = {'Name': ['Alice', 'Bob'], 'Age': [25, 30]}
df = pd.DataFrame(data)
print(df)

Choosing the Right Data Structure

Choosing the right data structure depends on the task:

• Use lists for simple collections of data.
• Use dictionaries for mapping and lookups.
• Use NumPy arrays or Pandas DataFrames for numerical and tabular data.

Common Mistakes in Using Data Structures

Here are some mistakes to avoid:

• Using lists for numerical computations instead of NumPy arrays.
• Not leveraging dictionaries for fast lookups.
• Failing to use Pandas for complex data manipulation tasks.

What Did We Learn?

In this lesson, you learned:

• The importance of data structures in data science.
• How to use Python's built-in data structures like lists, dictionaries, and tuples.
• How to work with NumPy arrays and Pandas DataFrames for advanced data handling.
• How to choose the appropriate data structure for specific tasks.

Great job! Let’s move to the next topic.