I want to extract a sublist from a nested list based on the first element in each list. I have the following nested list:
input = [
['nom', 'N', 'eye'],
['acc', 'E', 'computer'],
['dat', 'C', 'screen']
]
I want to have a function which returns ['nom', 'N', 'eye'] when the first element of the sublist 'nom' is inputted, for instance:
output = ['nom', 'N', 'eye'] # When the function takes 'nom' as its argument output = ['acc', 'E', 'computer'] # When the function takes 'acc' as its argument output = ['dat', 'C', 'screen'] # When the function takes 'dat' as its argument
How should I achieve this with python3.6+?
Advertisement
Answer
my_list = [
['nom', 'N', 'eye'],
['acc', 'E', 'computer'],
['dat', 'C', 'screen']
]
my_input = input("Enter first string to find: ")
for lis in my_list:
if lis[0] == my_input:
print(lis)
break
Output:
Enter variable name: nom ['nom', 'N', 'eye']