Python创建数组的坑

本文和大家分享的主要是python 中创建数组过程中的一些坑，一起来看看吧，希望对大家  学习python 有所帮助。

　　  1、问题的引出

　　在做leetcode 的题目时，遇到了一个求数组中出现次数最多的 k 个元素的题，参照答案的思路，首先利用一个 dict 记录所有元素出现的次数， key:value 中的 key 表示元素， value 表示元素出现的次数，随后根据元素出现的次数将元素放入对应的桶中，桶是一个二维数组，桶中第一个元素保存出现次数为 0 的元素，桶中第二个元素保存出现次数为 1 的元素，依次类推。最后从后往前遍历桶，取出出现次数最多的 k 个元素即可。

　　按照这样的思路，我写了如下的代码：

　　  class   Solution(object):

　　  def   topKFrequent(self, nums, k):

　　"""

　　:type nums: List[int]

　　:type k: int

　　:rtype: List[int]

　　"""

　　bucket1 = [[]] * (len(nums) + 1)

　　freq_map = dict()

　　  for num   in nums:

　　freq_map[num] = freq_map.get(num,0) + 1

　　  print (freq_map)

　　  for key   in freq_map:

　　#bucket[freq_map[key]].append(key)

　　bucket1[freq_map[key]].append(key)

　　res = []

　　  for i   in range(len(bucket1)-1,-1,-1):

　　res += bucket1 

　　if len(res) >= k:

　　return res[:k]

　　s = Solution()print (s.topKFrequent([4,1,-1,2,-1,2,3],2))

　　看似完美无缺啊，不过结果不对啊，结果输出了1和2，不应该是-1和2么，于是我打印了bucket1，发现是这样的：

　　[[1, 2, 3, 4, -1], [1, 2, 3, 4, -1], [1, 2, 3, 4, -1], [1, 2, 3, 4, -1], [1, 2, 3, 4, -1], [1, 2, 3, 4, -1], [1, 2, 3, 4, -1], [1, 2, 3, 4, -1]]

　　为什么所有的元素都一样呢？

　　于是 我换了一种方法创建bucket：

　　bucket = [[] for _ in range(len(nums)+1)]

　　结果正常，此时bucket的输出为：

　　[[], [1, 3, 4], [2, -1], [], [], [], [], []]

　　那么，看似一样的创建bucket的方法，为什么输出会不一样呢？

　　2、问题的解释

　　问题就在与，第一种即使用*创建数组时，数组中的每一个元素都是同一个list object，那么往任意一个中添加元素，实际上是往所有的元素中添加，所以会出现之前的结果，而用列表生成式的方法，不会有问题，因为每一个list都是不同的list object，我们来看一看下面的代码：

　　bucket = [[] for _ in range(len(nums)+1)]

bucket1 = [[]] * (len(nums) + 1)

print (bucket)print (bucket1)

print (id(bucket1[0]),id(bucket1[1]))

print(id(bucket[0]), id(bucket[1]))

　　输出为：

　　[[], [], [], [], [], [], [], []]

[[], [], [], [], [], [], [], []]

4342035336 4342035336

4342034568 4342034952

　　可以看到，bucket1中的每一个元素的id都是相同的，说明是同一个元素，而bucket中的每一个元素的id是不同的，说明是不同的元素。

来源：简书