# Before using NumPy, we need to install and import it.
 
import numpy as np
 
print('NumPy is ready!')

# Create a simple NumPy array
my_array = np.array([1, 2, 3, 4, 5])
print('Here is your array:', my_array)

# Arrays look like lists, but NumPy lets us do math on all elements at once.
result = my_array + 2
print('Add 2 to every value:', result)

# Subtract 1 from each value
print('Subtract 1:', my_array - 1)

# Multiply every value by 3
print('Multiply by 3:', my_array * 3)

# Divide all by 2
print('Divide by 2:', my_array / 2)

# You can also do math with two arrays of the same shape.
array_b = np.array([10, 20, 30, 40, 50])
sum_arrays = my_array + array_b
print('Add two arrays:', sum_arrays)

# Take the square root with np.sqrt
roots = np.sqrt(my_array)
print('Square roots:', roots)

# Raise every value to the power of 2
squared = np.power(my_array, 2)
print('Squares:', squared)

# Use input and math with arrays
user_num = int(input('Pick a whole number: '))
result = my_array * user_num
print('Your number times every value:', result)

# What happens if we use two arrays of different sizes?
array_short = np.array([1, 2])
try:
    print(my_array + array_short)
except ValueError as e:
    print('Error:', e)
    

# Find the sum, mean, min, and max.
print('Sum:', np.sum(my_array))
print('Mean (average):', np.mean(my_array))
print('Minimum:', np.min(my_array))
print('Maximum:', np.max(my_array))

# You can change an array element by index.
my_array[0] = 42
print('New array after change:', my_array)

# Select only values greater than 10
filtered = my_array[my_array > 10]
print('Values above 10:', filtered)

# Use array slicing to get only certain values.
first_two = my_array[:2]
print('First two values:', first_two)

# Loop through and print each value in my_array.
for value in my_array:
    print('Array element:', value)
    

# Use a list comprehension with arrays.
plus_five = [x + 5 for x in my_array]
print('Each value plus five:', plus_five)

# Ask the user for five test scores
scores = []
for i in range(5):
    s = float(input(f'Enter score {i + 1}: '))
    scores.append(s)
scores_array = np.array(scores)
print('Your scores:', scores_array)

# Calculate average, minimum, and maximum score
avg = np.mean(scores_array)
minimum = np.min(scores_array)
maximum = np.max(scores_array)
print('Average:', avg)
print('Lowest:', minimum)
print('Highest:', maximum)