# --------------
# Importing header files
import numpy as np

#Path of the file has been stored in variable called 'path'

# New record
new_record=[[50,  9,  4,  1,  0,  0, 40,  0]]

#Code starts here
data = np.genfromtxt(path,delimiter=",", skip_header=1)
census = np.concatenate((data,new_record),axis=0)

# print the concatenated Array
print(census)



# --------------
#Code starts here
age= census[:,0]

max_age= max(age)
min_age= min(age)
age_mean=age.mean()
age_std =age.std()


print ( "max age is: {0}, min age is:  {1}, mean age is: {2}, std age is {3}".format(max_age,min_age,age_mean,age_std))


# --------------
#Code starts here
race_0 = census[census[:,2] == 0]
print(race_0)

race_0 = census[census[:,2] == 0]
race_1 = census[census[:,2] == 1]
race_2 = census[census[:,2] == 2]
race_3 = census[census[:,2] == 3]
race_4 = census[census[:,2] == 4]

len_0=len(race_0)
len_1=len(race_1)
len_2=len(race_2)
len_3=len(race_3)
len_4=len(race_4)

print ( "len_0: {0}, len_1:  {1}, len_2: {2}, len_3: {3},len_4: {4}".format(len_0,len_1,len_2,len_3,len_4))

minority_race = 3


# --------------
#Code starts here
senior_citizens =census[census[:,0] > 60]
senior_citizens.shape

working_hours_sum = sum(senior_citizens[:,6])
print('working_hours_sum : ',working_hours_sum)

senior_citizens_len = len(senior_citizens)
print('senior_citizens_len : ',senior_citizens_len)

avg_working_hours = working_hours_sum/senior_citizens_len
print('avg_working_hours : ',avg_working_hours)


# --------------
#Code starts here

high =census[census[:,1] > 10]
low =census[census[:,1] <= 10]
high.shape,low.shape


avg_pay_high = high[:,7].mean()
avg_pay_low = low[:,7].mean()

print('avg_pay_high : ',avg_pay_high)
print('avg_pay_low : ',avg_pay_low)


