bites 118 - list and set

Author: [email protected]

118/duplicates.py

@@ -0,0 +1,15 @@
+def get_duplicate_indices(words):
+    """Given a list of words, loop through the words and check for each
+       word if it occurs more than once.
+       If so return the index of its first ocurrence.
+       For example in the following list 'is' and 'it'
+       occurr more than once, and they are at indices 0 and 1 so you would
+       return [0, 1]:
+       ['is', 'it', 'true', 'or', 'is', 'it', 'not?'] => [0, 1]
+       Make sure the returning list is unique and sorted in ascending order."""
+
+    # list of the words that occurs more than once
+    duplicated = [word for word in set(words) if words.count(word) > 1]
+    # list of first occurence position of duplicated words
+    pos = [words.index(dupword) for dupword in duplicated]
+    return sorted(pos)

118/test_duplicates.py

@@ -0,0 +1,23 @@
+from duplicates import get_duplicate_indices
+
+
+def test_get_duplicate_indices_docstring():
+    words = ['is', 'it', 'true', 'or', 'is', 'it', 'not']
+    assert get_duplicate_indices(words) == [0, 1]
+
+
+def test_get_duplicate_indices_bite_text():
+    words = ['this', 'is', 'a', 'new', 'bite', 'I', 'hope', 'this',
+             'bite', 'will', 'teach', 'you', 'something', 'new']
+    assert get_duplicate_indices(words) == [0, 3, 4]
+
+
+def test_get_duplicate_indices_another_text():
+    # keeping it simple with split on space, so lists != lists.
+    words = ('List comprehensions provide a concise way to create '
+             'lists. Common applications are to make new lists where '
+             'each element is the result of some operations applied '
+             'to each member of another sequence or iterable, or to '
+             'create a subsequence of those elements that satisfy a '
+             'certain condition').split()
+    assert get_duplicate_indices(words) == [3, 6, 7, 17, 22, 32]

140-pandas/medals.py

@@ -1,3 +1,5 @@
+from collections import namedtuple
+
 import pandas as pd
 
 data = "https://bites-data.s3.us-east-2.amazonaws.com/summer.csv"