COO format (Coordinate) in SciPy - Cheat Sheet & Quick Revision

Recall & Review

beginner

What does COO stand for in sparse matrix formats?

COO stands for Coordinate format. It stores a sparse matrix by listing the row and column coordinates of non-zero elements along with their values.

beginner

How does COO format store data internally?

COO format stores three arrays: one for row indices, one for column indices, and one for the non-zero values. Each position in these arrays corresponds to one non-zero element.

intermediate

Why is COO format useful for sparse matrices?

COO format is simple and efficient for constructing sparse matrices incrementally and for matrices with many non-zero elements scattered randomly.

beginner

How do you create a COO sparse matrix using scipy?

Use scipy.sparse.coo_matrix with three arrays: data, (row indices, column indices), and optionally the shape of the matrix.

intermediate

What is a limitation of COO format compared to CSR or CSC formats?

COO format is less efficient for arithmetic operations and matrix-vector products because it does not store data sorted by rows or columns.

Which arrays are stored in COO format for a sparse matrix?

A Only values

B Row pointers and values

C Row indices, column indices, and values

D Column pointers and values

COO format stores row indices, column indices, and the corresponding non-zero values.

What is the main advantage of COO format?

A Fast matrix multiplication

B Easy to build incrementally

C Sorted data by rows

D Uses less memory than CSR

COO format is simple and good for building sparse matrices incrementally.

Which scipy function creates a COO sparse matrix?

A scipy.sparse.coo_matrix

B scipy.sparse.csr_matrix

C scipy.sparse.dok_matrix

D scipy.sparse.lil_matrix

The function scipy.sparse.coo_matrix creates a COO format sparse matrix.

What is a drawback of COO format compared to CSR?

A Less efficient for arithmetic operations

B Cannot store zeros

C Requires more memory for dense matrices

D Does not support row indexing

COO format is less efficient for arithmetic and matrix-vector operations than CSR.

In COO format, how are the non-zero elements stored?

A Grouped by columns only

B Sorted by row and column

C Grouped by rows only

D In any order with matching row and column arrays

COO format stores non-zero elements in any order, with corresponding row and column indices.

Explain how COO format represents a sparse matrix and why it might be chosen over other formats.

Think about how you would list non-zero elements with their positions.

Describe the steps to create a COO sparse matrix using scipy and how to access its data.

Focus on the function and the arrays it needs.

Practice

1. What does the COO format in scipy primarily store?
coo_matrix is used to represent sparse matrices by storing which of the following?

easy

A. Only non-zero values and their row and column indices

B. All values including zeros in a dense matrix

C. Only the diagonal elements of a matrix

D. The matrix in a compressed row format

Solution

1. Step 1: Understand COO format basics and compare options

   COO format stores sparse matrices by keeping track of non-zero values and their positions. Only non-zero values and their row and column indices correctly describes storing non-zero values with their row and column indices.

2. Final Answer:

   Only non-zero values and their row and column indices -> Option A

3. Quick Check:

   COO = non-zero values + coordinates [OK]

Hint: COO stores non-zero values with positions only [OK]

Common Mistakes:

• Thinking COO stores all matrix values
• Confusing COO with compressed formats
• Assuming COO stores only diagonal elements

2. Which of the following is the correct way to create a COO sparse matrix from data arrays data, row, and col using scipy?

easy

A. coo_matrix((data, (row, col)))

B. coo_matrix(data, row, col)

C. coo_matrix(data, (row, col))

D. coo_matrix((row, col), data)

Solution

1. Step 1: Recall scipy coo_matrix syntax and check each option

   The correct syntax is coo_matrix((data, (row, col))) where data is values and (row, col) are coordinates. Only coo_matrix((data, (row, col))) matches the correct tuple structure required by coo_matrix.

2. Final Answer:

   coo_matrix((data, (row, col))) -> Option A

3. Quick Check:

   Tuple of (data, (row, col)) is correct [OK]

Hint: Use tuple: (data, (row, col)) for coo_matrix [OK]

Common Mistakes:

• Passing data, row, col as separate arguments
• Swapping order of data and coordinates
• Not using nested tuple for coordinates

3. What is the output of the following code?

from scipy.sparse import coo_matrix
row = [0, 1, 2]
col = [1, 2, 0]
data = [4, 5, 6]
coo = coo_matrix((data, (row, col)), shape=(3, 3))
print(coo.toarray())

medium

A. [[0 6 0]
[5 0 0]
[0 0 4]]

B. [[4 0 0]
[0 5 0]
[0 0 6]]

C. [[0 0 6]
[4 0 0]
[0 5 0]]

D. [[0 4 0]
[0 0 5]
[6 0 0]]

Solution

1. Step 1: Map data to matrix positions and construct full matrix

   Data 4 at (0,1), 5 at (1,2), 6 at (2,0) means matrix has these values at those coordinates. Positions without data are zero, so the matrix is [[0 4 0], [0 0 5], [6 0 0]].

2. Final Answer:

   [[0 4 0] [0 0 5] [6 0 0]] -> Option D

3. Quick Check:

   Data placed at (row, col) = matrix [OK]

Hint: Match data with (row, col) indices to build matrix [OK]

Common Mistakes:

• Mixing row and column indices
• Assuming data fills diagonal
• Confusing toarray() output formatting

4. Identify the error in this code snippet that tries to create a COO matrix:

from scipy.sparse import coo_matrix
row = [0, 1]
col = [1, 2]
data = [3]
coo = coo_matrix((data, (row, col)), shape=(3, 3))

medium

A. Shape parameter is missing

B. Length of data does not match length of row and col arrays

C. coo_matrix requires dense matrix input

D. row and col must be numpy arrays, not lists

Solution

1. Step 1: Check lengths of data, row, and col and understand coo_matrix requirements

   Data has length 1, but row and col have length 2, which is inconsistent. Data, row, and col must have the same length to map values correctly.

2. Final Answer:

   Length of data does not match length of row and col arrays -> Option B

3. Quick Check:

   Data, row, col lengths must match [OK]

Hint: Data, row, col arrays must be same length [OK]

Common Mistakes:

• Ignoring length mismatch errors
• Assuming lists are invalid input
• Missing shape parameter causes error

5. You have coordinate data for a large sparse matrix with many zero values. Which of the following is the best approach to efficiently create and store this matrix using scipy?

hard

A. Create a dense numpy array and convert it to coo_matrix

B. Store the matrix as a list of lists with zeros included

C. Use coo_matrix with data, row, and col arrays and convert to CSR format for fast arithmetic

D. Use a dictionary to store all matrix values including zeros

Solution

1. Step 1: Understand sparse matrix creation, optimize for operations, and evaluate options

   COO format efficiently stores non-zero values with coordinates, ideal for sparse data. Converting COO to CSR format improves arithmetic and slicing performance. Dense arrays waste memory; lists or dicts storing zeros are inefficient.

2. Final Answer:

   Use coo_matrix with data, row, and col arrays and convert to CSR format for fast arithmetic -> Option C

3. Quick Check:

   COO + CSR = efficient sparse matrix handling [OK]

Hint: Create COO then convert to CSR for speed [OK]

Common Mistakes:

• Using dense arrays wastes memory
• Storing zeros wastes space
• Ignoring format conversion for speed