主页 / 问题 / 849133
Accepted
Alexshev92
Asked:2020-07-02 16:53:57 +0000 UTC2020-07-02 16:53:57 +0000 UTC

将一个数组拆分为多个具有相同值的数组

  • 772

情况如下：假设有一些对象的数组；为简单起见，我们取字符数组。事先不知道里面有什么符号。您需要以某种方式将其分成几个仅包含相同字符的数组。例如

arr = ['a', 'b', 'a', 'c', 'b', 'd', 'c', 'a', 'd']

因此，输出应该是四个数组：

arr1 = ['a', 'a', 'a']
arr2 = ['b', 'b']
arr3 = ['c', 'c']
arr3 = ['d', 'd']

或者由这些数组组成的二维数组

arr = [['a', 'a', 'a'], ['b', 'b'], ['c', 'c'], ['d', 'd']]

尝试做这样的事情

arr = sorted(arr)
a = []
b = []
for i in range(1, len(arr)-1):
   if (arr[i] == arr[i+1]):
       a.append(arr[i])
else:
    b.append(a)
    a = []

然而，结果是

[['a'], ['b'], ['c']]

在这里，确切地说，需要的不是Python实现，而是对这个算法的理解，这似乎是一个简单的任务，但我就是不知道如何解决它

python

4 个回答

  1. Best Answer
    from collections import defaultdict

arr = ['a', 'b', 'a', 'c', 'b', 'd', 'c', 'a', 'd']
d = defaultdict(list)

for i in arr:
    d[i].append(i)

print(list(d.values()))  # [['a', 'a', 'a'], ['b', 'b'], ['c', 'c'], ['d', 'd']]

    如果不用于形成新的中间数据结构序列（字典），并理解一个简单算法的操作：

    arr = ['a', 'b', 'a', 'c', 'b', 'd', 'c', 'a', 'd']

result = []
map_dict = {}  # тут словарь не является промежуточной структурой данных
               # он служит лишь как карта для запоминания позиций списков 
               # в результирующем списке

for i in arr:
    if i in map_dict:
        pos = map_dict[i]
        result[pos].extend(i)
    else:
        pos = len(result)
        map_dict[i] = pos
        result.append([i])

print(result)  # [['a', 'a', 'a'], ['b', 'b'], ['c', 'c'], ['d', 'd']]
    • 4

  2. 尝试groupby：

    from itertools import groupby

arr = ['a', 'b', 'a', 'c', 'b', 'd', 'c', 'a', 'd']

arr_str = ''.join(sorted(arr))
print(arr_str)  # aaabbccdd

new_arr = [list(values) for _, values in groupby(arr_str)]
print(new_arr)  # [['a', 'a', 'a'], ['b', 'b'], ['c', 'c'], ['d', 'd']]
    • 3

  3. 如果我理解正确，那么你想要最容易理解的算法

    arr = ['a', 'b', 'a', 'c', 'b', 'd', 'c', 'a', 'd']
d = {}
for item in arr:
    if item in d:
       d[item].append(item)
    else:
        d[item] = [item]

print(d.values())
    • 2

  4. collections.Counter是专门为这个任务设计的：

    import collections

a = ['a', 'b', 'a', 'c', 'b', 'd', 'c', 'a', 'd']

for k, v in collections.Counter(a).items():
    print([k] * v)

    ['a', 'a', 'a']
['b', 'b']
['c', 'c']
['d', 'd']
    • 0

相关问题