Python: Extract the nth element from a given list of tuples using lambda

# Define a function 'extract_nth_element' that extracts the nth element from each tuple in a list
def extract_nth_element(test_list, n):
    # Use the 'map' function with a lambda to extract the nth element from each tuple in 'test_list'
    result = list(map(lambda x: x[n], test_list))
    
    # Return the resulting list
    return result

# Create a list of tuples 'students' containing names and two additional elements for each student
students = [
    ('Greyson Fulton', 98, 99),
    ('Brady Kent', 97, 96),
    ('Wyatt Knott', 91, 94),
    ('Beau Turnbull', 94, 98)
]

# Print the original list of tuples 'students'
print("Original list:")
print(students)

# Extract the 0th element from each tuple and print the result
n = 0
print("\nExtract nth element (n =", n, ") from the said list of tuples:")
print(extract_nth_element(students, n))

# Extract the 2nd element from each tuple and print the result
n = 2
print("\nExtract nth element (n =", n, ") from the said list of tuples:")
print(extract_nth_element(students, n))