#!/usr/bin/env python

# file: 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"]
xl.append("Joe")
xl.append("Bob")
print(xl)
xl2 = ["1.11", "2.22", "3.33"]
print(xl2)
xl3 = [1.11, 2.22, 3.33]
print(xl3)
xl2.append(float(1.11))
print("the new xl2: ", xl2)
xl2.append("Wow!")
print("the new and better xl2: ", xl2)

# 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)

# 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)

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

#
## 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:")
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)

# create a dictionary
#
a = dict(red=4139, green=4127, blue=4098)
print("--> dictionaries:")
print(a)

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

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

print("printing dictionary:")
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
#
for key in sorted(a):
    print(key, a[key])