如何在python中创建二维列表

Python012

如何在python中创建二维列表,第1张

Python中创建二维列表/数组，即创建一个list，并且这个list的元素还是list。可以用列表解析的方法实现。

创建例子如下：

2d_list = [[0 for col in range(cols)] for row in range(rows)]

其中cols, rows变量替换为你需要的数值即可，例如：

2d_list = [[0 for col in range(9)] for row in range(9)]

# 9*9的二维列表

如果 list 存储了若干复杂结构的值，比如这样的一个列表：

temp = [('a', 1, 1.5),

('b', 2, 5.1),

('c', 9, 4.3)]

你想找到其中是 ('b', XX, XX) 这样的元素，其中 XX 可以为任意值。这种情况无法通过 index 函数来获得，我们可以利用 sort 的 key 参数来实现。

list.sort（或者 sorted 函数）有一个 key 参数，你可以提供一个函数来作为排序的依据。此时我们可以传入以下值：

temp.sort(key = lambda x:x[0]!='b')

随后我们会发现，所有形如 ('b', XX, XX) 的元素全部浮动到了列表的头部，此时我们访问 temp[0] 就可以获得想要查找的值了。

我们也可以写一个简单的函数：

findindex = lambda self,i,value:sorted(self,key=lambda x:x[i]!=value)[0]

那我们只需要这样调用：

>>>findindex(temp,0,'b')

就会返回它找到的第一个值：

>>>('b',2)

如果你需要使用二维数组，在 Python 中，除了 numpy 这个包之外，最简单的方式就是使用 list 了。你可能认为可以这样构建：

但是请你再仔细想一想，这样做真的没有问题吗？

不要忘记了， Python 一切都是对象 这个问题，对象肯定存在引用的问题，尤其是可变对象。我们试着来改变其中一个元素：

很不幸，不是我们想象的那样，它改变了多个元素。

来看一看 [ [0] * m ] * n 这个表达式，它首先创建 a = [ [0] * 4] ，然后构造 array = [a] * 3 ，这里 array 里面是 a 的 3 个引用。

正确的构造，应该使用 列表推导式 ，而且非常 Pythonic ：