61 lines
1.5 KiB
Python
61 lines
1.5 KiB
Python
import re
|
|
import pandas as pd
|
|
|
|
|
|
def list_comprehension(l1: list):
|
|
"""For input list l1 of integers, return list of elements
|
|
whose square is less than sum of all elements"""
|
|
|
|
sum_l1 = sum(l1)
|
|
return [elem for elem in l1 if elem * elem < sum_l1]
|
|
|
|
|
|
def nested_list_comprehension(l1: list, l2: list):
|
|
"""For input lists l1, l2, return a list of elements common
|
|
to both l1 and l2."""
|
|
|
|
# result = []
|
|
# for e1 in l1:
|
|
# for e2 in l2:
|
|
# if e1 == e2:
|
|
# result.append(e1)
|
|
|
|
result = [e1 for e2 in l2 for e1 in l1 if e1 == e2]
|
|
|
|
return result
|
|
|
|
|
|
def dict_comprehension(l1: list):
|
|
"""For input list l1 of integers, return a dictionary with
|
|
keys from l1, and values are squares of keys"""
|
|
|
|
result = {e: e * e for e in l1}
|
|
|
|
return result
|
|
|
|
|
|
if __name__ == "__main__":
|
|
l1 = [1, 2, 3, 4]
|
|
l2 = [2, 4, 6, 8]
|
|
|
|
print("\n-------------------\n")
|
|
print("Inputs:\n")
|
|
print(f"l1: {l1}")
|
|
print(f"l2: {l2}")
|
|
print("\n-------------------\n")
|
|
|
|
print("list_comprehension:\n")
|
|
print(f"Desired output: [2, 4]")
|
|
print(f"Output : {list_comprehension(l2)}")
|
|
print("\n-------------------\n")
|
|
|
|
print("nested_list_comprehension:\n")
|
|
print(f"Desired output: [2, 4]")
|
|
print(f"Output : {nested_list_comprehension(l1, l2)}")
|
|
print("\n-------------------\n")
|
|
|
|
print("dict_comprehension:\n")
|
|
print("Desired output: {1: 1, 2: 4, 3: 9, 4: 16}")
|
|
print(f"Output: : {dict_comprehension(l1)}")
|
|
print("\n-------------------\n")
|