# -*- coding: utf-8 -*-
"""
Created on Fri Nov 11 10:57:01 2022

@author: cpresser
"""
import numpy as np

x = np.linspace(-5, 10, 20)
print(x)

print("-----------------------")

avgX = np.average(x)
print(f"The average is: {avgX}.")

print("-----------------------")

y = 2*x + 1
print(y)

print("-----------------------")
print(np.add(x,y))

print("-----------------------")
print(x+y)

print("-----------------------")
# linspace for 2d
array2d = np.linspace([0, 0], [10, -10], 5)
print(array2d)

print("-----------------------")
#print all values of array2d, one per line, 
# using nested loops
for x in array2d:
    for i in x:
        print(i)
        
print("-----------------------")
#loop through the indices
for i in range(len(array2d)):
    for j in range(len(array2d[i])):
        print(array2d[i,j], end=" ")
    print()
    

print("-----------------------")

#go through each value and print them
for v in np.nditer(array2d):
    print(v)

print("-----------------------")

#go through indices and values
for index, value in np.ndenumerate(array2d):
    print(f"index: {index} value: {value}")
    # index[0] and index[1] hold the index values
    # array2d [index[0]] [index[1]]
    # array2d [index[0], index[1]]
    # array2d [index]
    array2d[index] *= 2

print("-----------------------")
print(array2d)

print("-----------------------")

array3d = np.zeros((2, 3, 4))
print(array3d)

print("-----------------------")
#enumerate through a 3d array
for index, value in np.ndenumerate(array3d):
    array3d[index] = index[0] + index[1] + index[2]

print(array3d)

print("-----------------------")
#use nested loops
#for each index in the first dimension (2d array)
for i in range(len(array3d)):
    #for each index in the second dim (1d array)
    for j in range(len(array3d[i])):
        #for each index in the 3rd dim (0d array)
        for k in range(len(array3d[i][j])):
            array3d[i,j,k] = i + j + k
  
print(array3d)