欢迎加入QQ讨论群258996829
苹果6袋
麦子学院
Python数组创建中的坑详解
发布时间：2017-08-16 17:22 回复:0 查看:2400 最后回复：2017-08-16 17:22
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是不同的，说明是不同的元素。
来源：简书