#Tutorial 2 Solutions

#Section 1.1
name = 'Imad Pasha'
print name[0:4]
full_name = name[0:4] + ' thebest ' + name[5:]

#Section 1.2
import numpy as np
arr = np.arange(1,101)
arr = 2*arr
print arr
arr[5] = 0
arr = np.append(arr, 42)
print 'Max: ', np.max(arr)
print 'Min: ', np.min(arr)
print 'Mean: ',np.mean(arr)
arr = arr[::-1]

#Section 3.1
spectrum = np.loadtxt('neon_spec.txt')
spectrum = np.transpose(spectrum)
pixels = spectrum[0]
signal = spectrum[1] #Note: don't reverse pixels, they are in order
signal = signal[::-1]
truncated_pixels = pixels[501:]
truncated_signal = signal[501:]
normed_signal = truncated_signal / np.mean(truncated_signal)

import matplotlib.pyplot as plt
plt.plot(truncated_pixels, normed_signal)
plt.title("A Neon Spectrum")
plt.xlabel("Pixel Number")
plt.ylabel("Signal [adu]")
plt.show()


#Homework (should be a different script)
import numpy as np
a = raw_input('Enter leading coefficient (a): ')
a = float(a)
b = raw_input("Enter second coefficient (b): ")
b = float(b)
c = raw_input("Enter final coefficient (c): ")
c = float(c)

discriminant = b**2 - 4*a*c
denominator = 2*a
x1 = ((-b) + np.sqrt(discriminant)) / denominator
x2 = ((-b) - np.sqrt(discriminant)) / denominator

print "The roots of your polynomial are: ", x1, x2

#Bonus Hw (probably wont be needed)

import numpy as np
a = raw_input("Enter leading coefficient (a): ")
a = float(a)
b = raw_input("Enter second coefficient (b): ")
b = float(b)
c = raw_input("Enter final coefficient (c): ")
c = float(c)

discriminant = b**2 - 4*a*c
denominator = 2*a
if (discriminant > 0) or (discriminant == 0):
	x1 = ((-b) + np.sqrt(discriminant)) / denominator
	x2 = ((-b) - np.sqrt(discriminant)) / denominator
	print "The roots of your polynomial are: ", x1, x2
elif discriminant < 0:
	print 'Your Polynomial has no real roots'