Dictionaries¶
[1]:
# Colab setup ------------------
import os, sys, subprocess
if "google.colab" in sys.modules:
cmd = "pip install --upgrade watermark"
process = subprocess.Popen(cmd.split(), stdout=subprocess.PIPE)
# ------------------------------
Dictionaries, which is Python’s built-in mapping object. A mapping object allows an arbitrary collection of objects to be indexed by an arbitrary collection of values. That’s a mouthful. It is easier to understand instead by comparing to a sequence.
Let’s take a sequence of two strings, say a tuple containing a first and last name.
name = ('jeffrey', 'lebowski')
We are restricted on how we reference the sequence. I.e., the first name is name[0]
, and the last name is name[1]
. A mapping object could instead be indexed like name['first name']
and name['last name']
. You can imagine this would be very useful! A classic example in biology might be looking up amino acids that are coded for by given codons. E.g., you might want
aa['CTT']
to give you 'Leucine'
.
Python’s only build-in mapping type is a dictionary. You might imagine that the Oxford English Dictionary might conveniently be stored as a dictionary (obviously). I.e., you would not want to store definitions that have to be referenced like
oed[103829]
Rather, you would like to get definitions like this:
oed['computer']
Dictionary syntax¶
The syntax for creating a dictionary, as usual, is best seen through example.
[2]:
my_dict = {'a': 6, 'b': 7, 'c': 27.6}
my_dict
[2]:
{'a': 6, 'b': 7, 'c': 27.6}
A dictionary is created using curly braces ({}
). Each entry has a key, followed by a colon, and then the value associated with the key. In the example above, the keys are all strings, which is the most common use case. Note that the items can be of any type; in the above example, they are int
s and a float
.
We could also create the dictionary using the built-in dict()
function, which can take a tuple of 2-tuples, each one containing a key-value pair.
[3]:
dict((('a', 6), ('b', 7), ('c', 27.6)))
[3]:
{'a': 6, 'b': 7, 'c': 27.6}
Finally, we can make a dictionary with keyword arguments to the dict()
function.
[4]:
dict(a='yes', b='no', c='maybe')
[4]:
{'a': 'yes', 'b': 'no', 'c': 'maybe'}
We do not need to have strings as the keys. In fact, any immutable object can be a key.
[5]:
my_dict = {
0: 'zero',
1.7: [1, 2, 3],
(5, 6, 'dummy string'): 3.14,
'strings are immutable': 42
}
my_dict
[5]:
{0: 'zero',
1.7: [1, 2, 3],
(5, 6, 'dummy string'): 3.14,
'strings are immutable': 42}
However, mutable objects cannot be used as keys.
[6]:
my_dict = {
'immutable is ok': 1,
['mutable', 'not', 'ok']: 5
}
---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
<ipython-input-6-ccfe6ce7f575> in <module>
----> 1 my_dict = {
2 'immutable is ok': 1,
3 ['mutable', 'not', 'ok']: 5
4 }
TypeError: unhashable type: 'list'
A dictionary use case in bioinformatics¶
It might be useful to quickly look up 3-letter amino acid codes. Dictionaries are particularly useful for this.
[7]:
aa_dict = {
"A": "Ala",
"R": "Arg",
"N": "Asn",
"D": "Asp",
"C": "Cys",
"Q": "Gln",
"E": "Glu",
"G": "Gly",
"H": "His",
"I": "Ile",
"L": "Leu",
"K": "Lys",
"M": "Met",
"F": "Phe",
"P": "Pro",
"S": "Ser",
"T": "Thr",
"W": "Trp",
"Y": "Tyr",
"V": "Val",
}
Another useful dictionary would contain the set of codons and the amino acids they code for. This is built in the code below using the zip()
function we learned before. To see the logic on how this is constructed, see the codon table here.
[8]:
# The set of DNA bases
bases = ['T', 'C', 'A', 'G']
# Build list of codons
codon_list = []
for first_base in bases:
for second_base in bases:
for third_base in bases:
codon_list += [first_base + second_base + third_base]
# The amino acids that are coded for (* = STOP codon)
amino_acids = 'FFLLSSSSYY**CC*WLLLLPPPPHHQQRRRRIIIMTTTTNNKKSSRRVVVVAAAADDEEGGGG'
# Build dictionary from tuple of 2-tuples (technically an iterator, but it works)
codons = dict(zip(codon_list, amino_acids))
# Show that we did it
codons
[8]:
{'TTT': 'F',
'TTC': 'F',
'TTA': 'L',
'TTG': 'L',
'TCT': 'S',
'TCC': 'S',
'TCA': 'S',
'TCG': 'S',
'TAT': 'Y',
'TAC': 'Y',
'TAA': '*',
'TAG': '*',
'TGT': 'C',
'TGC': 'C',
'TGA': '*',
'TGG': 'W',
'CTT': 'L',
'CTC': 'L',
'CTA': 'L',
'CTG': 'L',
'CCT': 'P',
'CCC': 'P',
'CCA': 'P',
'CCG': 'P',
'CAT': 'H',
'CAC': 'H',
'CAA': 'Q',
'CAG': 'Q',
'CGT': 'R',
'CGC': 'R',
'CGA': 'R',
'CGG': 'R',
'ATT': 'I',
'ATC': 'I',
'ATA': 'I',
'ATG': 'M',
'ACT': 'T',
'ACC': 'T',
'ACA': 'T',
'ACG': 'T',
'AAT': 'N',
'AAC': 'N',
'AAA': 'K',
'AAG': 'K',
'AGT': 'S',
'AGC': 'S',
'AGA': 'R',
'AGG': 'R',
'GTT': 'V',
'GTC': 'V',
'GTA': 'V',
'GTG': 'V',
'GCT': 'A',
'GCC': 'A',
'GCA': 'A',
'GCG': 'A',
'GAT': 'D',
'GAC': 'D',
'GAA': 'E',
'GAG': 'E',
'GGT': 'G',
'GGC': 'G',
'GGA': 'G',
'GGG': 'G'}
A dictionary is an implementation of a hash table¶
It is useful to stop and think about how a dictionary works. Let’s create a dictionary and look at where the values are stored in memory.
[9]:
# Create dictionary
my_dict = {'a': 6, 'b': 7, 'c':12.6}
# Find where they are stored
print(id(my_dict))
print(id(my_dict['a']))
print(id(my_dict['b']))
print(id(my_dict['c']))
140482662461504
4364753408
4364753440
140482662789552
So, each entry in the dictionary is stored at a different location in memory. The dictionary itself also has its own address. So, when I index a dictionary with a key, how does the dictionary know which address in memory to use to fetch the value I am interested in?
Dictionaries use a hash function to do this. A hash function converts its input to an integer. For example, we can use Python’s built-in hash function to convert the keys to integers.
[10]:
hash('a'), hash('b'), hash('c')
[10]:
(-3286243246506062721, -4637256982569203119, 4044209238390845933)
Under the hood, Python then converts these integers to integers that could correspond to locations in memory. A collection of elements that can be indexed this way is called a hash table. This is a very common data structure in computing. Wikipedia has a pretty good discussion on them.
Given what you know about how dictionaries work, why do you think mutable objects are not acceptable as keys?
Dictionaries are mutable¶
Dictionaries are mutable. This means that they can be changed in place. For example, if we want to add an element to a dictionary, we use simple syntax.
[11]:
# Remind ourselves what the dictionary is
print(my_dict)
# Add an entry
my_dict['d'] = 'Data analysis is so much fun!'
# Look at dictionary again
print(my_dict)
# Change an entry
my_dict['a'] = 'I was not satisfied with entry a.'
# Look at it again
print(my_dict)
{'a': 6, 'b': 7, 'c': 12.6}
{'a': 6, 'b': 7, 'c': 12.6, 'd': 'Data analysis is so much fun!'}
{'a': 'I was not satisfied with entry a.', 'b': 7, 'c': 12.6, 'd': 'Data analysis is so much fun!'}
Membership operators with dictionaries¶
The in
and not in
operators work with dictionaries, but both only query keys and not values. We see this again by example.
[12]:
# Make a fresh my_dict
my_dict = {'a': 1, 'b': 2, 'c': 3}
# in works with keys
'b' in my_dict, 'd' in my_dict, 'e' not in my_dict
[12]:
(True, False, True)
[13]:
# Try it with values
2 in my_dict
[13]:
False
Yup! We get False
. Why? Because 2
is not a key in my_dict
. We can also iterate over the keys in a dictionary using the in
keyword.
[14]:
for key in my_dict:
print(key, ':', my_dict[key])
a : 1
b : 2
c : 3
The best, and preferred, method, is to iterate over key
,value
pairs in a dictionary using the items()
method of a dictionary.
[15]:
for key, value in my_dict.items():
print(key, value)
a 1
b 2
c 3
Note, however, that like lists, the item
s that come out of the my_dict.items()
iterator are not items in the dictionary, but copies of them. If you make changes within the for
loop, you will not change entries in the dictionary.
[16]:
for key, value in my_dict.items():
value = 'this string will not be in dictionary.'
my_dict
[16]:
{'a': 1, 'b': 2, 'c': 3}
You will, though, if you use the keys.
[17]:
for key, _ in my_dict.items():
my_dict[key] = 'this will be in the dictionary.'
my_dict
[17]:
{'a': 'this will be in the dictionary.',
'b': 'this will be in the dictionary.',
'c': 'this will be in the dictionary.'}
Built-in functions for dictionaries¶
The built-in len()
function and del
operation work on dictionaries.
len(d)
gives the number of entries in dictionaryd
del d[k]
deletes entry with keyk
from dictionaryd
This is the first time we’ve encountered the del
keyword. This keyword is used to delete variables and their values from memory. The del
keyword can also be to delete items from lists. Let’s see things in practice.
[18]:
# Create my_list and my_dict for reference
my_dict = dict(a=1, b=2, c=3, d=4)
my_list = [1, 2, 3, 4]
# Print them
print('my_dict:', my_dict)
print('my_list:', my_list)
# Get lengths
print('length of my_dict:', len(my_dict))
print('length of my_list:', len(my_list))
# Delete a key from my_dict
del my_dict['b']
# Delete entry from my_list
del my_list[1]
# Show post-deleted objects
print('post-deleted my_dict:', my_dict)
print('post-deleted my_list:', my_list)
my_dict: {'a': 1, 'b': 2, 'c': 3, 'd': 4}
my_list: [1, 2, 3, 4]
length of my_dict: 4
length of my_list: 4
post-deleted my_dict: {'a': 1, 'c': 3, 'd': 4}
post-deleted my_list: [1, 3, 4]
Note, though, that you cannot delete an item from a tuple, since it’s immutable.
[19]:
my_tuple = (1, 2, 3, 4)
del my_tuple[1]
---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
<ipython-input-19-db6fdb3f0eac> in <module>
1 my_tuple = (1, 2, 3, 4)
----> 2 del my_tuple[1]
TypeError: 'tuple' object doesn't support item deletion
Dictionary methods¶
Dictionaries have several built-in methods in addition to the items()
you have already seen. Following are a few of them, assuming the dictionary is d
.
method |
effect |
---|---|
|
return keys |
|
return value associated with |
|
return the values in |
Let’s try these out.
[20]:
my_dict = dict(a=1, b=2, c=3, d=4)
my_dict.keys()
[20]:
dict_keys(['a', 'b', 'c', 'd'])
Note that this is a dict_keys
object. We cannot index it. If, say, we wanted to sort the keys and have them index-able, we would have to convert them to a list.
[21]:
sorted(list(my_dict.keys()))
[21]:
['a', 'b', 'c', 'd']
This is not a usual use case, though, and be warned that doing then when this is not explicitly what you want can lead to bugs. Now let’s try popping an entry out of the dictionary.
[22]:
my_dict.pop('c')
[22]:
3
[23]:
my_dict
[23]:
{'a': 1, 'b': 2, 'd': 4}
…and, as we expect, key 'c'
is now deleted, and its value was returned in the call to my_dict.pop('c')
. Finally, we can look at the values.
[24]:
my_dict.values()
[24]:
dict_values([1, 2, 4])
We get a dict_values
object, similar to the dict_keys
object we got with the my_dict.keys()
method.
You can get more information about build-in methods from the Python documentation.
Dictionary comprehensions¶
Like lists could be created using list comprehensions, so too can dictionaries be created using dictionary comprehensions. Recall from when we made a dictionary of codon/amino acid codes before that we build them using
[25]:
codons = dict(zip(codon_list, amino_acids))
As a reminder, here are the first five entries of codon_list
and amino_acids
.
[26]:
print('codon list:', codon_list[:5])
print('amino_acids:', amino_acids[:5])
codon list: ['TTT', 'TTC', 'TTA', 'TTG', 'TCT']
amino_acids: FFLLS
We could create the same dictionary codons
with a dictionary comprehension as follows.
[27]:
codons = {key: value for key, value in zip(codon_list, amino_acids)}
A list comprehension works analogously to a list comprehension. The only differences are that the dictionary comprehension is enclosed in curly braces and each entry is given by a key and value pair (as opposed to a single entry for a list comprehension) using the syntax key: value
. (The names of the variables can be anything, e.g., k: v
is legal.) Of course, the iterators in the dictionary comprehension also need to emit both keys and values as well.
Using dictionaries as kwargs¶
A nifty feature of dictionaries is that they can be passed into functions as keyword arguments. We covered named keyword arguments in the lesson on functions. In addition to the named keyword arguments, a function can take in arbitrary keyword arguments (not arbitrary non-keyword arguments). This is specified in the function definition by including a last argument with a double-asterisk, **
. The kwargs with the double-asterisk get passed in as a dictionary.
[28]:
def concatenate_sequences(a, b, **kwargs):
"""Concatenate (combine) 2 or more sequences."""
seq = a + b
for key in kwargs:
seq += kwargs[key]
return seq
Let’s try it!
[29]:
concatenate_sequences('TGACAC', 'CAGGGA', c='GGGGGGGGG', d='AAAATTTTT')
[29]:
'TGACACCAGGGAGGGGGGGGGAAAATTTTT'
Now, imagine we have a dictionary that contains our values.
[30]:
my_dict = {
'a': 'TGACAC',
'b': 'CAGGGA',
'c': 'GGGGGGGGG',
'd': 'AAAATTTTT'
}
We can now pass this directly into the function by preceding it with a double asterisk.
[31]:
concatenate_sequences(**my_dict)
[31]:
'TGACACCAGGGAGGGGGGGGGAAAATTTTT'
Beautiful! This example is kind of trivial, but you can imagine that it can come in handy, e.g. with large sets of sequence fragments that you read in from a file. Many packages, particularly plotting packages, use **kwargs
extensively.
Computing environment¶
[32]:
%load_ext watermark
%watermark -v -p jupyterlab
CPython 3.8.3
IPython 7.16.1
jupyterlab 2.1.5