Commit 79df0eff authored by Stephen Pearce's avatar Stephen Pearce Committed by Maxim Belkin

code/readings_05.py: Harmonise with lesson text (#748)

parent 30229abb
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)
assert action in ['--min', '--mean', '--max'], \
'Action is not one of --min, --mean, or --max: ' + action
for filename in filenames:
process(filename, action)
def process(filename, action):
data = numpy.loadtxt(filename, delimiter=',')
if action == '--min':
values = data.min(axis=1)
values = numpy.min(data, axis=1)
elif action == '--mean':
values = data.mean(axis=1)
values = numpy.mean(data, axis=1)
elif action == '--max':
values = data.max(axis=1)
for m in values:
print(m)
values = numpy.max(data, axis=1)
for val in values:
print(val)
if __name__ == '__main__':
main()
main()
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment