要提取列表的一列,可以使用列表解析或者循環來完成。
使用列表解析:
lst = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
column = [row[1] for row in lst]
print(column) # 輸出 [2, 5, 8]
使用循環:
lst = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
column = []
for row in lst:
column.append(row[1])
print(column) # 輸出 [2, 5, 8]
在以上兩種方法中,lst
是待提取列的列表,row[1]
表示取每個子列表的第二個元素,并將其加入到新的列 column
中。