Python列表維度轉換及二維列表排序

一、二維轉一維

（1）使用sum函數進行轉換：

intervals = [[1,4],[2,5],[3,6]]
a = sum(intervals,[])
print(a)

輸出結果：

[1, 4, 2, 5, 3, 6]

（2）使用列表推導式

intervals = [[1,4],[2,5],[3,6]]
a = [i for j in intervals for i in j]
print(a)

輸出結果：

[1, 4, 2, 5, 3, 6]

二、一維轉二維

循環實現：

a = [1, 4, 2, 5, 3, 6]
s = []
if (len(a) % 2 == 0):
    for i in range(0, len(a), 2):
        s.append([a[i], a[i + 1]])
print(s)

輸出結果：

[[1, 4], [2, 5], [3, 6]]

三、二維列表排序

按子列表中的第一個元素排序：

a = [[1, 4],[3, 6],[4, 7],[2, 5]]
a = sorted(a,key=(lambda x:x[0]))
print(a)

 

輸出結果：

[[1, 4], [2, 5], [3, 6], [4, 7]]

 

按子列表中的第二個元素排序：

a = [[1, 4],[3, 3],[4, 7],[2, 5]]
a = sorted(a,key=(lambda x:x[1]))
print(a)

 

輸出結果：

[[3, 3], [1, 4], [2, 5], [4, 7]]