String type in NumPy - Mini Project: Build & Apply

String type in NumPy

📖 Scenario: You work in a small company that stores customer names in a list. You want to use NumPy to handle these names efficiently for future analysis.

🎯 Goal: You will create a NumPy array of customer names using the string data type, then filter names based on length, and finally display the filtered names.

📋 What You'll Learn

Create a NumPy array with string type

Use a variable to set the minimum length for filtering

Use a list comprehension to filter names by length

Print the filtered names

💡 Why This Matters

🌍 Real World

Handling customer or user names efficiently is common in data science for cleaning and preparing data.

💼 Career

Knowing how to work with string data in NumPy helps in data preprocessing tasks in many data science and analytics jobs.

1

Create a NumPy array of customer names

Import NumPy as np and create a NumPy array called names with these exact strings: 'Alice', 'Bob', 'Charlie', 'David', 'Eva'. Make sure the array uses the string data type 'U'.

NumPy

# Import numpy and create the names array with string type
# Your code here

Hint

Use np.array() with dtype='U' to create a string array.

2

Set the minimum length for filtering

Create a variable called min_length and set it to 5. This will be the minimum number of characters a name must have to be selected.

NumPy

import numpy as np
names = np.array(['Alice', 'Bob', 'Charlie', 'David', 'Eva'], dtype='U')
# Create min_length variable
# Your code here

Hint

Just assign the number 5 to min_length.

3

Filter names by minimum length

Use a list comprehension to create a list called filtered_names that contains only the names from names whose length is greater than or equal to min_length. Use for name in names and len(name) to check the length.

NumPy

import numpy as np
names = np.array(['Alice', 'Bob', 'Charlie', 'David', 'Eva'], dtype='U')
min_length = 5
# Filter names by length using list comprehension
# Your code here

Hint

Use [name for name in names if len(name) >= min_length] to filter.

4

Print the filtered names

Print the variable filtered_names to display the list of names that meet the length condition.

NumPy

import numpy as np
names = np.array(['Alice', 'Bob', 'Charlie', 'David', 'Eva'], dtype='U')
min_length = 5
filtered_names = [name for name in names if len(name) >= min_length]
# Print the filtered names
# Your code here

Hint

Use print(filtered_names) to show the result.

Practice

1. Which NumPy string type should you use to store Unicode text like emojis or accented characters?

easy

A. Use 'U' type for Unicode strings.

B. Use 'S' type for Unicode strings.

C. Use 'i' type for Unicode strings.

D. Use 'f' type for Unicode strings.

Solution

1. Step 1: Understand NumPy string types

   NumPy has two main string types: 'S' for byte strings (ASCII) and 'U' for Unicode strings.

2. Step 2: Identify correct type for Unicode

   Unicode strings require 'U' type to support characters beyond ASCII, like emojis or accented letters.

3. Final Answer:

   Use 'U' type for Unicode strings. -> Option A

4. Quick Check:

   Unicode text = 'U' type [OK]

Hint: Unicode needs 'U', ASCII uses 'S' type [OK]

Common Mistakes:

• Using 'S' type for Unicode causes data loss
• Confusing numeric types with string types
• Assuming 'U' is for byte strings

2. Which of the following is the correct way to create a NumPy array of byte strings with length 5?

easy

A. np.array(['hello', 'world'], dtype='int5')

B. np.array(['hello', 'world'], dtype='U5')

C. np.array(['hello', 'world'], dtype='S5')

D. np.array(['hello', 'world'], dtype='float5')

Solution

1. Step 1: Identify byte string dtype syntax

   Byte strings use dtype='S' followed by the fixed length, e.g., 'S5' for length 5.

2. Step 2: Check options for correct dtype

   np.array(['hello', 'world'], dtype='S5') uses 'S5', which is correct for byte strings of length 5.

3. Final Answer:

   np.array(['hello', 'world'], dtype='S5') -> Option C

4. Quick Check:

   Byte strings = 'S' + length [OK]

Hint: Byte strings use dtype='S' plus length number [OK]

Common Mistakes:

• Using 'U' instead of 'S' for byte strings
• Using invalid dtypes like 'int5' or 'float5'
• Omitting the length number after 'S'

3. What will be the output of this code?

import numpy as np
arr = np.array(['cat', 'dog', 'elephant'], dtype='S5')
print(arr)

medium

A. ['cat' 'dog' 'eleph']

B. [b'cat' b'dog' b'eleph']

C. [b'cat' b'dog' b'elephant']

D. ['cat' 'dog' 'elephant']

Solution

1. Step 1: Understand dtype='S5' effect

   Using dtype='S5' means fixed-length byte strings of length 5, so strings longer than 5 characters get truncated.

2. Step 2: Check each string length and truncation

   'cat' and 'dog' fit within 5 bytes, but 'elephant' is longer and gets truncated to 'eleph'.

3. Final Answer:

   [b'cat' b'dog' b'eleph'] -> Option B

4. Quick Check:

   Fixed length 5 truncates 'elephant' to 'eleph' [OK]

Hint: Fixed-length strings truncate longer text [OK]

Common Mistakes:

• Expecting full 'elephant' without truncation
• Confusing byte strings with Unicode strings
• Ignoring the 'b' prefix for byte strings

4. The following code throws an error. What is the main issue?

import numpy as np
arr = np.array(['a', 'b', 'c'], dtype='U2')
arr[0] = 'abc'

medium

A. The array must be created with dtype='S3' to assign 'abc'.

B. You cannot assign strings to NumPy arrays.

C. dtype='U2' only supports byte strings, not Unicode.

D. The string 'abc' is longer than dtype='U2' allows.

Solution

1. Step 1: Understand dtype='U2' limitation

   dtype='U2' means Unicode strings of fixed length 2 characters only.

2. Step 2: Check assigned string length

   Trying to assign 'abc' (3 characters) to an element that only allows 2 characters causes truncation or error.

3. Final Answer:

   The string 'abc' is longer than dtype='U2' allows. -> Option D

4. Quick Check:

   Fixed-length Unicode strings truncate or error on longer input [OK]

Hint: Check string length vs dtype length before assignment [OK]

Common Mistakes:

• Thinking dtype='U2' supports any length string
• Believing NumPy arrays can't hold strings
• Confusing byte strings ('S') with Unicode ('U')

5. You have a list of mixed-length Unicode strings: ['hi', 'hello', 'hey', 'hola']. You want to create a NumPy array that stores all strings without truncation. Which dtype should you choose?

hard

A. dtype='U5' because the longest string has 5 characters.

B. dtype='S5' because it supports Unicode strings up to length 5.

C. dtype='U4' because it is the default for Unicode strings.

D. dtype='S10' to allow extra space for all strings.

Solution

1. Step 1: Find the longest string length

   The longest string is 'hello' with 5 characters.

2. Step 2: Choose dtype to fit longest string

   Use dtype='U5' to store Unicode strings up to length 5 without truncation.

3. Step 3: Avoid byte string dtype for Unicode

   Using 'S' type stores byte strings, not Unicode, so it may cause data loss.

4. Final Answer:

   dtype='U5' because the longest string has 5 characters. -> Option A

5. Quick Check:

   Unicode max length = dtype='U' + max length [OK]

Hint: Match dtype length to longest Unicode string [OK]

Common Mistakes:

• Using 'S' dtype for Unicode strings
• Choosing dtype length smaller than longest string
• Assuming default dtype fits all strings