#!/usr/bin/env python

# file: lecture_30/example.py
#
# demonstrate some basic data structures
#

# include 
#
import os
import sys
import numpy as np

# create a list, which can be thought of as an array
#
print("1D array:")

xl = ["Alex", "Jordan"]
print(xl)

xl.append("Joe")
xl.append("Bob")
print(xl)

print(xl[0])
print(xl[1])

xl2 = ["1.11", "2.22", "3.33"]
print(xl2)

xl3 = [1.11, 2.22, 3.33]
print(xl3)

xl.append(int(27))
xl.append(float(27.0))
print("xl:", xl)

xl.append(xl2)
print("x1:", xl)


# do some more lists
#
xl3 = [1.11, 2.22, 3.33]
print(xl3)
xl2 = xl3
xl2.append(float(1.11))
print("the new xl2: ", xl2)
xl2.append("Wow!")
print("the new and better xl2: ", xl2)

# let's go back to basic Python
#
zz = [[1.11, 2.22, 3.33], "Joe", [4.44, 5.55]]
print(zz)
#

# create a 1D numeric array
#
print("now try this with numpy:")
x = np.array([1.11, 2.22, 3.33])
print(x)

print("a true 32-bit float array:")
y = np.array([1.11, 2.22, 3.33], np.float32)
print(y)

z = float(23.23)
print(z)

# create a 2D numeric array
#
print("2D array:")
z = np.array([[1.11, 2.22, 3.33], [4.44, 5.55, 6.66]], np.float32)
print(z)

# create a list
#
print("---> list:")
l = ["alex", "jordan"]
print("list: ", l)
l.append("mike")
print("append: ", l)
#
l.append(float(27.27))
print ("append: ", l)
l.append("daniel")
l.append(float(35.01))
print ("append: ", l)
##
#l = [["joe", float(3.33)], ["mary", float(2.0)]]
print("heterogeneous: ", l)
l.append(float(35.01))
print("heterogeneous: ", l)
l.append([27.0, 99.0])
l.append("Alex")
print("heterogeneous: ", l)

# loop over a list
#
print("--> looping:")
l = ["aaa", [1.0, 2.0, 3.0], "bbb", [4.0], "ccc"]
for element in l:
    print(element)
print("--> end of looping")

#
print("--> indexed looping:")
for i in range(0, len(l)):
    print(i, l[i])

print("--> better looping:")
l = [1, 2, 3.77, 4, 5]
for arg in l:
    print("arg = ", arg)

# list comprehension
#
l = [1, 2, 3, 4, 5]
squares = [x**2 for x in l]
print("--> squares: ", squares)

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

# create a dictionary
#
a = dict(red=4139, green=4127, blue=4098)
print("--> dictionaries:")
print("orig:", a)
print("specific value", a['red'])

a['white'] = ["Joe", 313131]
print("new a:", a)

a['red'] = ["Alex", "Jordan"]
print("overwrite:", a)

a["mary"] = float(27.27)
a["joe"] = ["car", "house", float(27.27)]
print("new version:", a)
a['dict'] = dict(yellow=27, white=2727)
print("--> better value of a:")
print(a)

print("printing dictionary:")
print("---->:")
for key in a:
    print(" ->", key, a[key])

print("another dictionary print:")
for v, w in enumerate(a):
    print(">>> index:", v, "key:", w, "[", a[w], "]")

# ordered dictionaries are now the default
#
#a = dict(red=4139, green=4127, blue=4098)
#print(a)

# sort a dictionary
#
print("---->:")
for key in a:
    print(key, a[key])
print("----sorted>:")
b = sorted(a)
print(b)
for key in sorted(a):
    print(key, a[key])