Adding required files

Author: ggmarshall

code/argv_list.py

@@ -0,0 +1,2 @@
+import sys
+print('sys.argv is', sys.argv)

code/arith.py

@@ -0,0 +1,34 @@
+import sys
+
+
+def main():
+    assert len(sys.argv) == 4, 'Need exactly 3 arguments'
+
+    operator = sys.argv[1]
+    assert operator in ['add', 'subtract', 'multiply', 'divide'], (
+        'Operator is not one of add, subtract, multiply, or divide: '
+        'bailing out')
+    try:
+        operand1, operand2 = float(sys.argv[2]), float(sys.argv[3])
+    except ValueError:
+        print('Cannot convert input to a number: bailing out')
+        return
+
+    do_arithmetic(operand1, operator, operand2)
+
+
+def do_arithmetic(operand1, operator, operand2):
+
+    if operator == 'add':
+        value = operand1 + operand2
+    elif operator == 'subtract':
+        value = operand1 - operand2
+    elif operator == 'multiply':
+        value = operand1 * operand2
+    elif operator == 'divide':
+        value = operand1 / operand2
+    print(value)
+
+
+if __name__ == '__main__':
+    main()

code/check.py

@@ -0,0 +1,35 @@
+import sys
+import numpy
+
+
+def main():
+    script = sys.argv[0]
+    filenames = sys.argv[1:]
+    if len(filenames) <= 1:  # nothing to check
+        print('Only 1 file specified on input')
+    else:
+        nrow0, ncol0 = row_col_count(filenames[0])
+        print('First file %s: %d rows and %d columns' % (
+            filenames[0], nrow0, ncol0))
+        for f in filenames[1:]:
+            nrow, ncol = row_col_count(f)
+            if nrow != nrow0 or ncol != ncol0:
+                print('File %s does not check: %d rows and %d columns'
+                      % (f, nrow, ncol))
+            else:
+                print('File %s checks' % f)
+        return
+
+
+def row_col_count(filename):
+    try:
+        nrow, ncol = numpy.loadtxt(filename, delimiter=',').shape
+    except ValueError:
+        # This occurs if the file doesn't have same number of rows and columns,
+        # or if it has non-numeric content
+        nrow, ncol = (0, 0)
+    return nrow, ncol
+
+
+if __name__ == '__main__':
+    main()

code/count_stdin.py

@@ -0,0 +1,7 @@
+import sys
+
+count = 0
+for line in sys.stdin:
+    count += 1
+
+print(count, 'lines in standard input')

code/gen_inflammation.py

@@ -0,0 +1,20 @@
+#!/usr/bin/env python
+
+"""
+Generate pseudo-random patient inflammation data for use in Python lessons.
+"""
+
+import random
+
+n_patients = 60
+n_days = 40
+n_range = 20
+
+middle = n_days / 2
+
+for p in range(n_patients):
+    vals = []
+    for d in range(n_days):
+        upper = max(n_range - abs(d - middle), 0)
+        vals.append(random.randint(upper/4, upper))
+    print(','.join([str(v) for v in vals]))

code/line_count.py

@@ -0,0 +1,40 @@
+import sys
+
+
+def main():
+    """
+    print each input filename and the number of lines in it,
+    and print the sum of the number of lines
+    """
+    filenames = sys.argv[1:]
+    sum_nlines = 0  # initialize counting variable
+
+    if len(filenames) == 0:  # no filenames, just stdin
+        sum_nlines = count_file_like(sys.stdin)
+        print('stdin: %d' % sum_nlines)
+    else:
+        for f in filenames:
+            n = count_file(f)
+            print('%s %d' % (f, n))
+            sum_nlines += n
+        print('total: %d' % sum_nlines)
+
+
+def count_file(filename):
+    """count the number of lines in a file"""
+    f = open(filename, 'r')
+    nlines = len(f.readlines())
+    f.close()
+    return(nlines)
+
+
+def count_file_like(file_like):
+    """count the number of lines in a file-like object (eg stdin)"""
+    n = 0
+    for line in file_like:
+        n = n+1
+    return n
+
+
+if __name__ == '__main__':
+    main()

code/my_ls.py

@@ -0,0 +1,18 @@
+import glob
+import sys
+
+
+def main():
+    """prints names of all files with sys.argv as suffix"""
+    assert len(sys.argv) >= 2, "Argument list cannot be empty"
+    # NB: behaviour is not as you'd expect if sys.argv[1] is *
+    suffix = sys.argv[1]
+    glob_input = '*.' + suffix  # construct the input
+    glob_output = glob.glob(glob_input)  # call the glob function
+    for item in glob_output:  # print the output
+        print(item)
+    return
+
+
+if __name__ == '__main__':
+    main()

code/readings_01.py

@@ -0,0 +1,10 @@
+import sys
+import numpy
+
+
+def main():
+    script = sys.argv[0]
+    filename = sys.argv[1]
+    data = numpy.loadtxt(filename, delimiter=',')
+    for m in data.mean(axis=1):
+        print(m)

code/readings_02.py

@@ -0,0 +1,14 @@
+import sys
+import numpy
+
+
+def main():
+    script = sys.argv[0]
+    filename = sys.argv[1]
+    data = numpy.loadtxt(filename, delimiter=',')
+    for m in data.mean(axis=1):
+        print(m)
+
+
+if __name__ == '__main__':
+    main()

code/readings_03.py

@@ -0,0 +1,14 @@
+import sys
+import numpy
+
+
+def main():
+    script = sys.argv[0]
+    for filename in sys.argv[1:]:
+        data = numpy.loadtxt(filename, delimiter=',')
+        for m in data.mean(axis=1):
+            print(m)
+
+
+if __name__ == '__main__':
+    main()

code/readings_04.py

@@ -0,0 +1,25 @@
+import sys
+import numpy
+
+
+def main():
+    script = sys.argv[0]
+    action = sys.argv[1]
+    filenames = sys.argv[2:]
+
+    for f in filenames:
+        data = numpy.loadtxt(f, delimiter=',')
+
+        if action == '--min':
+            values = data.min(axis=1)
+        elif action == '--mean':
+            values = data.mean(axis=1)
+        elif action == '--max':
+            values = data.max(axis=1)
+
+        for m in values:
+            print(m)
+
+
+if __name__ == '__main__':
+    main()

code/readings_05.py

@@ -0,0 +1,30 @@
+import sys
+import numpy
+
+
+def main():
+    script = sys.argv[0]
+    action = sys.argv[1]
+    filenames = sys.argv[2:]
+    assert action in ['--min', '--mean', '--max'], (
+        'Action is not one of --min, --mean, or --max: ' + action)
+    for f in filenames:
+        process(f, action)
+
+
+def process(filename, action):
+    data = numpy.loadtxt(filename, delimiter=',')
+
+    if action == '--min':
+        values = data.min(axis=1)
+    elif action == '--mean':
+        values = data.mean(axis=1)
+    elif action == '--max':
+        values = data.max(axis=1)
+
+    for m in values:
+        print(m)
+
+
+if __name__ == '__main__':
+    main()

code/readings_06.py

@@ -0,0 +1,33 @@
+import sys
+import numpy
+
+
+def main():
+    script = sys.argv[0]
+    action = sys.argv[1]
+    filenames = sys.argv[2:]
+    assert action in ['--min', '--mean', '--max'], (
+        'Action is not one of --min, --mean, or --max: ' + action)
+    if len(filenames) == 0:
+        process(sys.stdin, action)
+    else:
+        for f in filenames:
+            process(f, action)
+
+
+def process(filename, action):
+    data = numpy.loadtxt(filename, delimiter=',')
+
+    if action == '--min':
+        values = data.min(axis=1)
+    elif action == '--mean':
+        values = data.mean(axis=1)
+    elif action == '--max':
+        values = data.max(axis=1)
+
+    for m in values:
+        print(m)
+
+
+if __name__ == '__main__':
+    main()

code/readings_07.py

@@ -0,0 +1,33 @@
+import sys
+import numpy
+
+
+def main():
+    script = sys.argv[0]
+    action = sys.argv[1]
+    filenames = sys.argv[2:]
+    assert action in ['-n', '-m', '-x'], (
+        'Action is not one of -n, -m, or -x: ' + action)
+    if len(filenames) == 0:
+        process(sys.stdin, action)
+    else:
+        for f in filenames:
+            process(f, action)
+
+
+def process(filename, action):
+    data = numpy.loadtxt(filename, delimiter=',')
+
+    if action == '-n':
+        values = data.min(axis=1)
+    elif action == '-m':
+        values = data.mean(axis=1)
+    elif action == '-x':
+        values = data.max(axis=1)
+
+    for m in values:
+        print(m)
+
+
+if __name__ == '__main__':
+    main()

code/readings_08.py

@@ -0,0 +1,41 @@
+import sys
+import numpy
+
+
+def main():
+    script = sys.argv[0]
+    if len(sys.argv) == 1:  # no arguments, so print help message
+        print('Usage: python readings_08.py action filenames\n'
+              'action must be one of --min --mean --max\n'
+              'if filenames is blank, input is taken from stdin;\n'
+              'otherwise, each filename in the list of arguments\n'
+              'is processed in turn')
+        return
+
+    action = sys.argv[1]
+    filenames = sys.argv[2:]
+    assert action in ['--min', '--mean', '--max'], (
+        'Action is not one of --min, --mean, or --max: ' + action)
+    if len(filenames) == 0:
+        process(sys.stdin, action)
+    else:
+        for f in filenames:
+            process(f, action)
+
+
+def process(filename, action):
+    data = numpy.loadtxt(filename, delimiter=',')
+
+    if action == '--min':
+        values = data.min(axis=1)
+    elif action == '--mean':
+        values = data.mean(axis=1)
+    elif action == '--max':
+        values = data.max(axis=1)
+
+    for m in values:
+        print(m)
+
+
+if __name__ == '__main__':
+    main()