Python Lists and List Manipulation

In Python, lists store an ordered collection of items which can be of different types. This post is an overview of lists and their manipulation.



By Michael Galarnyk, Data Scientist


Python Lists and List Manipulation Video

 

Before starting, I should mention that the code in this blog post and in the video above is available on my github.

 

Defining a List

 
Lists are written within square brackets []

Figure

Defining a List. The second row in this table index is how you access items in the list.

 

# Define a list
z = [3, 7, 4, 2]


Lists store an ordered collection of items which can be of different types. The list defined above has items that are all of the same type (int), but all the items of a list do not need to be of the same type as you can see below.

# Define a list
heterogenousElements = [3, True, 'Michael', 2.0]


The list contains an int, a bool, a string, and a float.

 

Access Values in a List

 
Each item in a list has an assigned index value. It is important to note that python is a zero indexed based language. All this means is that the first item in the list is at index 0.

Figure

Access item at index 0 (in blue)

 

# Define a list
z = [3, 7, 4, 2]# Access the first item of a list at index 0
print(z[0])


Figure

Output of accessing the item at index 0.

 

Python also supports negative indexing. Negative indexing starts from the end. It can be more convenient at times to use negative indexing to get the last item in the list because you don’t have to know the length of the list to access the last item.

Figure

Accessing the item at the last index.

 

# print last item in the list
print(z[-1])


Figure

Output of accessing the last item in the List

 

As a reminder, you could also access the same item using positive indexes (as seen below).

Figure

Alternative way of accessing the last item in the list z

 

 

Slice of Lists

 
Slices are good for getting a subset of values in your list. For the example code below, it will return a list with the items from index 0 up to and not including index 2.

Figure

First index is inclusive (before the :) and last (after the :) is not

 

# Define a list
z = [3, 7, 4, 2]print(z[0:2])


Figure

Slice of a list syntax.

 

# everything up to but not including index 3
print(z[:3])


The code below returns a list with items from index 1 to the end of the list

# index 1 to end of list
print(z[1:])


 

Update Item in a List

 

Lists in Python are mutable. All that means is that after defining a list, it is possible to update the individual items in a list.

# Defining a list
z = [3, 7, 4, 2]# Update the item at index 1 with the string "fish"
z[1] = "fish"
print(z)


Figure

Code to modify an item in a list

 

 

List Methods

 
Python lists have different methods that help you modify a list. This section of the tutorial just goes over various python list methods.

 

Index Method

 

# Define a list
z = [4, 1, 5, 4, 10, 4]


The index method returns the first index at which a value occurs. In the code below, it will return 0.

print(z.index(4))


You can also specify where you start your search.

print(z.index(4, 3))


 

Count Method

 
The count method works just like how it sounds. It counts the number of times a value occurs in a list

random_list = [4, 1, 5, 4, 10, 4]
random_list.count(5)


 

Sort Method

 

Figure

Sort a Python List - the actual code would be: z.sort()

 

The sort method sorts and alters the original list in place.

z = [3, 7, 4, 2]
z.sort()
print(z)


The code above sorts a list from low to high. The code below shows that you can also sort a list from high to low.

Figure

Sort a python list from high to low

 

# Sorting and Altering original list
# high to low
z.sort(reverse = True)
print(z)


As an aside, I should mention that you can also sort a list of strings from a-z and z-a as you can see here.

 

Append Method

 

Figure

Add the value 3 to the end of the list.

 

The append method adds an element to the end of a list. This happens inplace.

z = [7, 4, 3, 2]
z.append(3)
print(z)


 

Remove Method

 

The remove method removes the first occurrence of a value in a list.

z = [7, 4, 3, 2, 3]
z.remove(2)
print(z)


Figure

Code removes the first occurrence of the value 2 from the list z

 

 

Pop Method

 

Figure

z.pop(1) removes the value at index 1 and returns the value 4.

 

The pop method removes an item at the index you provide. This method will also return the item you removed from the list. If you don’t provide an index, it will by default remove the item at the last index.

z = [7, 4, 3, 3]
print(z.pop(1))
print(z)


 

Extend Method

 

The method extends a list by appending items. The benefit of this is you can add lists together.

z = [7, 3, 3]
z.extend([4,5])
print(z)


Figure

Add the list [4, 5] to the end of the list z.

 

Alternatively, the same thing could be accomplished by using the + operator.

print([1,2] + [3,4])


 

Insert Method

 

Figure

insert the list [1,2] at index 4

 

The insert method inserts an item before the index you provide

z = [7, 3, 3, 4, 5]
z.insert(4, [1, 2])
print(z)


 

Closing Remarks

 
Please let me know if you have any questions either here or in the comments section of the youtube video or through Twitter! Next post reviews python dictionaries. If you want to learn how to utilize the Pandas, Matplotlib, or Seaborn libraries, please consider taking my Python for Data Visualization LinkedIn Learning course. Here is a free preview video.

 
Bio: Michael Galarnyk is a Data Scientist and Corporate Trainer. He currently works at Scripps Translational Research Institute. You can find him on Twitter (https://twitter.com/GalarnykMichael), Medium (https://medium.com/@GalarnykMichael), and GitHub (https://github.com/mGalarnyk).

Original. Reposted with permission.

Related: