要提取列表的一列,可以使用列表解析或者循環(huán)來(lái)完成。
使用列表解析:
lst = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
column = [row[1] for row in lst]
print(column) # 輸出 [2, 5, 8]
使用循環(huán):
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]
表示取每個(gè)子列表的第二個(gè)元素,并將其加入到新的列 column
中。