python about linking sublist with same number together

To group the sublists in a general way you can:

Code:

def linking_sublists(lists):
    index = {}
    sets = []
    for l in lists:
        found = None
        for i in l:
            if i in index:
                # this list has an element we have already seen
                if found:
                    # combine two sets
                    to_remove = index[i]
                    if found != to_remove:
                        for j in index[i]:
                            found.add(j)
                            index[j] = found
                        to_remove.clear()
                else:
                    found = index[i]

        if found is not None:
            s = found
            for i in l:
                s.add(i)
        else:
            s = set(l)
            sets.append(s)
        for i in l:
            index[i] = s

    return [list(sorted(s)) for s in sets if s]

How:

This function uses sets and an index dict to group any list with matching elements into sets and track which elements are already in a set.

Test Code:

list_2 = [[2, 3], [4, 3], [6, 5], [7, 6], [7, 8], [13, 14], [30, 32]]
print(linking_sublists(list_2))

list_3 = [[2, 3], [4, 3], [6, 5], [7, 6], [7, 8], [30, 32], [4, 5], [3, 4]]
print(linking_sublists(list_3))

Results:

[[2, 3, 4], [5, 6, 7, 8], [13, 14], [30, 32]]
[[2, 3, 4, 5, 6, 7, 8], [30, 32]]

Related Contents:

  1. How to split a string into a list?
  2. How do I concatenate two lists in Python?
  3. Transpose list of lists
  4. How to sort a list of objects based on an attribute of the objects?
  5. Remove all occurrences of a value from a list?
  6. Remove all the elements that occur in one list from another
  7. Convert [key1,val1,key2,val2] to a dict?
  8. Element-wise addition of 2 lists?
  9. How to check if all elements of a list match a condition?
  10. Pythonic way to check if a list is sorted or not
  11. Python pandas insert list into a cell
  12. Python’s most efficient way to choose longest string in list?
  13. Finding max value in the second column of a nested list?
  14. Get list from pandas dataframe column or row?
  15. Finding median of list in Python
  16. How to find all positions of the maximum value in a list?
  17. list to dictionary conversion with multiple values per key?
  18. List on python appending always the same value [duplicate]
  19. Is there a better way to iterate over two lists, getting one element from each list for each iteration? [duplicate]
  20. In Python, what is the fastest algorithm for removing duplicates from a list so that all elements are unique *while preserving order*? [duplicate]
  21. Unable to reverse lists in Python, getting “Nonetype” as list
  22. Split list into smaller lists (split in half)
  23. Sum of list of lists; returns sum list
  24. Python – How to check list monotonicity
  25. How can I initialize a dictionary of distinct empty lists in Python?
  26. How can I convert each item in the list to string, for the purpose of joining them? [duplicate]
  27. How do I prepend to a short python list?
  28. How to check if an object is a list or tuple (but not string)?
  29. str.startswith with a list of strings to test for
  30. Why do two identical lists have a different memory footprint?
  31. How to split strings inside a list by whitespace characters
  32. Perform a string operation for every element in a Python list
  33. How to make a set of lists
  34. How do I merge multiple lists into one list? [duplicate]
  35. How to allow list append() method to return the new list
  36. convert csv file to list of dictionaries
  37. How to multiply all integers inside list [duplicate]
  38. Complexity of list.index(x) in Python
  39. Select 50 items from list at random
  40. List as a member of a python class, why is its contents being shared across all instances of the class?
  41. Find min, max, and average of a list
  42. Why does list ask about __len__?
  43. How do I extract all the values of a specific key from a list of dictionaries?
  44. What’s the difference between lists enclosed by square brackets and parentheses in Python?
  45. Co-occurrence matrix from nested list of words
  46. Joining elements in a list without the join command
  47. Union find implementation using Python
  48. list comprehension with multiple conditions (python)
  49. Converting string to tuple without splitting characters
  50. Column of lists, convert list to string as a new column

Leave a Comment