matrixA = [['AAA', 'BBB', 'CCC'], ['PPP', 'QQQ', 'RRR', 'SSS'], ['DDD','EEE','FFF'], ['GGG', 'HHH', 'III']]
how to remove the above elements from listA from matrixA
listA = ['DDD', 'EEE', 'FFF']
Desired output:
output_matrix = [['DDD','EEE','FFF'], ['GGG', 'HHH', 'III']]
sorry for my bad language
Advertisement
Answer
You can get the index
of listA
within matrixA
and then slice with it:
idx = matrixA.index(listA) out = matrixA[idx:]
to get
>>> out [["DDD", "EEE", "FFF"], ["GGG", "HHH", "III"]]
in case listA
doesn’t exist, it gives an error so we try
:
try: idx = matrixA.index(listA) except ValueError: # if it doesn't exist.. out = matrixA # we can take the whole `matrixA`, for example else: # it does exist out = matrixA[idx:]