2.8 Bonus: Loops
Teaching: 40 min || Exercises: 10 min
Overview
2.8.1 Background
Loops are a programming construct which allow us to repeat a command or set of commands for each item in a list. As such they are key to productivity improvements through automation. Similar to wildcards and tab completion, using loops also reduces the amount of typing required (and hence reduces the number of typing mistakes).
Suppose we have several hundred genome data files named basilisk.dat
, minotaur.dat
, and unicorn.dat
. For this example, we’ll use the 02_unix_intro/creatures
sub-directory which only has three example files, but the principles can be applied to many more files at once.
The structure of these files is the same: the common name, classification, and updated date are presented on the first three lines, with DNA sequences on the following lines. Let’s look at the files:
head -n 5 basilisk.dat minotaur.dat unicorn.dat
We would like to print out the classification for each species, which is given on the second line of each file. For each file, we would need to execute the command head -n 2
and pipe this to tail -n 1
. We’ll use a loop to solve this problem, but first let’s look at the general form of a loop.
2.8.2 The for
loop
The general synthax for the for
loop is:
for thing in list_of_things
do
operation_using $thing # Indentation within the loop is not required, but aids legibility
done
and we can apply this to our example like this:
When the shell sees the keyword for
, it knows to repeat a command (or group of commands) once for each item in a list. Each time the loop runs (called an iteration), an item in the list is assigned in sequence to the variable, and the commands inside the loop are executed, before moving on to the next item in the list. Inside the loop, we call for the variable’s value by putting $
in front of it. The $
tells the shell interpreter to treat the variable as a variable name and substitute its value in its place, rather than treat it as text or an external command.
In this example, the list is three filenames: basilisk.dat
, minotaur.dat
, and unicorn.dat
. Each time the loop iterates, it will assign a file name to the variable filename
and run the head
command. The first time through the loop, $filename
is basilisk.dat
. The interpreter runs the command head
on basilisk.dat
and pipes the first two lines to the tail
command, which then prints the second line of basilisk.dat
. For the second iteration, $filename
becomes minotaur.dat
. This time, the shell runs head
on monotaur.dat
and pipes the first two lines to the tail
command, which then prints the second line of monotaur.dat
. For the third iteration, $filename
becomes unicorn.dat
, so the shell runs the head
command on that file, and tail
on the output of that. Since the list was only three items, the shell exits the for
loop.
When using variables it is also possible to put the names into curly braces to clearly delimit the variable name: $filename
is equivalent to ${filename}
, but is different from ${file}name
. You may find this notation in other people’s programs.
We have called the variable in this loop filename
in order to make its purpose clearer to human readers. The shell itself doesn’t care what the variable is called; if we wrote this loop as:
for x in basilisk.dat minotaur.dat unicorn.dat
do
head -n 2 $x | tail -n 1
done
or:
for temperature in basilisk.dat minotaur.dat unicorn.dat
do
head -n 2 $temperature | tail -n 1
done
it would work exactly the same way.
Don’t do this. Programs are only useful if people can understand them, so meaningless names (like x
) or misleading names (like temperature
) increase the odds that the program won’t do what its readers think it does.
In the above examples, the variables (thing, filename, x and temperature) could have been given any other name, as long as it is meaningful to both the person writing the code and the person reading it.
Also note that loops can be used for other things than filenames, like a list of numbers or a subset of data.
Let’s continue with our example in the 02_unix_intro/creatures
sub-directory. Here’s a slightly more complicated loop:
for filename in *.dat
do
echo $filename
head -n 100 $filename | tail -n 20
done
The shell starts by expanding *.dat
to create the list of files it will process. The loop body then executes two commands for each of those files. The first command, echo
, prints its command-line arguments to standard output. For example:
In this case, since the shell expands $filename
to be the name of a file, echo $filename
prints the name of the file. Note that we can’t write this as:
for filename in *.dat
do
$filename
head -n 100 $filename | tail -n 20
done
because then the first time through the loop, when $filename
expanded to basilisk.dat
, the shell would try to run basilisk.dat
as a program. Finally, the head
and tail
combination selects lines 81-100 from whatever file is being processed (assuming the file has at least 100 lines).
Dealing with Spaces in Names
Spaces are used to separate the elements of the list that we are going to loop over. If one of those elements contains a space character, we need to surround it with quotes, and do the same thing to our loop variable. Suppose our data files are named:
red dragon.dat
purple unicorn.dat
To loop over these files, we would need to add double quotes like this:
for filename in "red dragon.dat" "purple unicorn.dat"
do
head -n 100 "$filename" | tail -n 20
done
It is simpler to avoid using spaces (or other special characters) in filenames.
The files above don’t exist, so if we run the above code, the head
command will be unable to find them, however the error message returned will show the name of the files it is expecting:
backing up files with cp
We would like to modify each of the files in 02_unix_intro/creatures
, but also save a version of the original files, naming the copies original-basilisk.dat
and original-unicorn.dat
. We can’t use:
cp *.dat original-*.dat
because that would expand to:
cp basilisk.dat minotaur.dat unicorn.dat original-*.dat
This wouldn’t back up our files, instead we get an error:
This problem arises when cp
receives more than two inputs. When this happens, it expects the last input to be a directory where it can copy all the files it was passed. Since there is no directory named original-*.dat
in the creatures
directory we get an error.
Instead, we can use a loop:
for filename in *.dat
do
cp $filename original-$filename
done
This loop runs the cp
command once for each filename. The first time, when $filename
expands to basilisk.dat
, the shell executes:
cp basilisk.dat original-basilisk.dat
The second time, the command is:
cp minotaur.dat original-minotaur.dat
The third and last time, the command is:
cp unicorn.dat original-unicorn.dat
Since the cp
command does not normally produce any output, it’s hard to check that the loop is doing the correct thing. However, we learned earlier how to print strings using echo
, and we can modify the loop to use echo
to print our commands without actually executing them. As such we can check what commands would be run in the unmodified loop.
The following diagram shows what happens when the modified loop is executed, and demonstrates how the judicious use of echo
is a good debugging technique.
Nelle’s Pipeline: Processing Files
Nelle is now ready to process her data files using goostats
— a shell script written by her supervisor. This calculates some statistics from a protein sample file, and takes two arguments:
- an input file (containing the raw data)
- an output file (to store the calculated statistics)
Since she’s still learning how to use the shell, she decides to build up the required commands in stages. Her first step is to make sure that she can select the right input files — remember, these are ones whose names end in ‘A’ or ‘B’, rather than ‘Z’. Starting from her home directory, Nelle types:
Her next step is to decide what to call the files that the goostats
analysis program will create. Prefixing each input file’s name with ‘stats’ seems simple, so she modifies her loop to do that:
She hasn’t actually run goostats
yet, but now she’s sure she can select the right files and generate the right output filenames.
Typing in commands over and over again is becoming tedious, though and Nelle is worried about making mistakes, so instead of re-entering her loop, she presses the up arrow. In response, the shell redisplays the whole loop on one line (using semi-colons to separate the pieces):
for datafile in NENE*[AB].txt; do echo $datafile stats-$datafile; done
Using the left arrow key, Nelle backs up and changes the command echo
to bash goostats
:
for datafile in NENE*[AB].txt; do bash goostats $datafile stats-$datafile; done
When she presses Enter, the shell runs the modified command. However, nothing appears to happen — there is no output. After a moment, Nelle realizes that since her script doesn’t print anything to the screen any longer, she has no idea whether it is running, much less how quickly. She kills the running command by typing Ctrl-C
, uses up-arrow to repeat the command, and edits it to read:
for datafile in NENE*[AB].txt; do echo $datafile; bash goostats $datafile stats-$datafile; done
When she runs her program now, it produces one line of output every five seconds or so:
1518 times 5 seconds, divided by 60, tells her that her script will take about two hours to run. As a final check, she opens another terminal window, goes into north-pacific-gyre/2012-07-03
, and uses cat stats-NENE01729B.txt
to examine one of the output files. It looks good, so she decides to get some coffee and catch up on her reading.
:::::
Performing a Dry Run
A loop is a way to do many things at once — or to make many mistakes at once if it does the wrong thing. One way to check what a loop would do is to echo
the commands it would run instead of actually running them. Suppose we want to preview the commands the following loop will execute without actually running those commands:
for file in *.pdb
do
analyze $file > analyzed-$file
done
Nested loops
2.8.3 Credit
Information on this page has been adapted and modified from the following source(s):
Gabriel A. Devenyi (Ed.), Gerard Capes (Ed.), Colin Morris (Ed.), Will Pitchers (Ed.),Greg Wilson, Gerard Capes, Gabriel A. Devenyi, Christina Koch, Raniere Silva, Ashwin Srinath, … Vikram Chhatre. (2019, July). swcarpentry/shell-novice: Software Carpentry: the UNIX shell, June 2019 (Version v2019.06.1).
https://github.com/cambiotraining/UnixIntro
https://github.com/cambiotraining/unix-shell